我在Tensorflow-gpu 1.9.0
上使用Ubuntu 18.04
,GPU驱动程序为410
,CUDA版本为10
。
我必须对需要边界框和相应图像名称的对象检测结果进行后期处理。
我知道我可以使用下面的函数来获取坐标并将其转储到文件中,但是我也想要文件名。
def draw_bounding_boxes_on_image(image,
boxes,
color='red',
thickness=4,
display_str_list_list=()):
我还尝试使用PIL数据结构来获取文件名,方法是
image.filename
个参数,但是没有存储。
我正在使用model_main.py文件进行训练,但是它不接受任何参数来将坐标转储到text / json / xml文件中。
如果有人有解决方案,请分享。