У меня есть коллекция изображений и соответствующих текстовых файлов, составляющих мой набор данных, которые я помещаю парами в набор данных тензорного потока. В текстовых файлах есть строки в формате:
<class> <x> <y> <w> <h>
Я хочу использовать API тензорного потока для преобразования моего текстового файла в данные, которые можно использовать для обучения, но сохраняйте только строки, где класс равен 0 До сих пор у меня были идеи, как этого добиться. Один из них - преобразовать RaggedTensor, в котором теперь находится переменная метки, в DenseTensor, поскольку формат файла известен и не меняется. Но я не нашел способа сделать это. Другая идея - использовать что-то вроде tf.where и tf.gather, но безуспешно.
Мой код на данный момент:
label = tf.io.read_file(file_path)
label = tf.strings.split(label, sep="\n")
label = tf.strings.to_number(tf.strings.split(label))
label = # filter lines that does not have class 0