Блог

Итак, после запуска этой строки кода
python -m scripts.label_image --graph=tf_files/retrained_graph.pb --image=tf_files/106906.jpg

моя консоль выдает вывод, подобный этому:

Время обработки (1-изображение): 0,294 с
ClassA (оценка = 0,97202)
ClassB (оценка = 0,02572)
ClassC (оценка = 0,00226)

Есть ли способ, которым я могу поместить выходные данные в формат .txt или CSV? Я попытался добавить простой>output.txt в конец предыдущей команды, но это создало только пустой текстовый файл.

Исходный код выглядит следующим образом:

 for i in top_k:
   print(template.format(labels[i], results[i]))
  

Который я изменил на этот:

 for i in top_k:
    outputFile = open('output.txt', 'w')
    print(template.format(labels[i], results[i]), file = outputFile)
    outputFile.close()
  

К сожалению, это печатает только последнее. По сути, он печатает первое сообщение, а затем помещает второе вместо первого и т.д. Есть ли способ сохранить все строки в этом output.txt файл?

Вы могли бы либо распечатать выходные данные в файл с помощью print (3.x).

Пример (из статьи выше):

 # Code for printing to a file 
sample = open('samplefile.txt', 'w') 
  
print('GeeksForGeeks', file = sample) 
sample.close()
  

Или, если вы работаете на bash, перенаправьте sys.stdout в file. Или для Windows.

В качестве альтернативы вы могли бы извлечь его по атрибутам, например, с помощью понимания списка, и использовать CSV-библиотеку python для сохранения его в файл.