建议paddledetection中得infer配置代码规范书写
收藏
如题:建议infer配置代码规范书写
ppdet/engine/trainer.py Trainer中这样书写:
def predict(self,
images,
draw_threshold=0.5,
output_dir='output',
save_txt=False):
self.dataset.set_images(images)
loader = create('TestReader')(self.dataset, 0)
imid2path = self.dataset.get_imid2path()
anno_file = self.dataset.get_anno()
clsid2catid, catid2name = get_categories(self.cfg.metric, anno_file)
反映到配置文件中,会这样:
anno_path: label_list.txt
即anno_path才是label_list
label_list反而没用到不起作用。。。。。。
飞桨_PaddleGAN
已解决
2#
回复于2021-09
您好,对于上面的书写规范我们有pre-commit会做出限制 ;关于label_list,在predict阶段,使用的ImageFolder加载数据,里面只有anno_path这个字段,label_list也是从这个字段读取的,这里我们会在文档中加强下说明~ 感谢您使用PaddleDetection~
0
收藏
请登录后评论
您好,对于上面的书写规范我们有pre-commit会做出限制 ;关于label_list,在predict阶段,使用的ImageFolder加载数据,里面只有anno_path这个字段,label_list也是从这个字段读取的,这里我们会在文档中加强下说明~ 感谢您使用PaddleDetection~
嗯嗯,是我当时想当然了,从字面意思理解了,没认真看
anno_path在predict阶段其实是label_list
emmm,确实让人容易产生歧义