如何在Tensorflow对象检测API中访问visualization_utils.py文件中的图像名称?

时间:2019-02-26 16:56:06

标签: tensorflow object-detection-api

我在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文件中。

如果有人有解决方案,请分享。

0 个答案:

没有答案