YOLOv7——save-txt参数路径更改

Question

我用YOLOv7 ready模型做对象检测。python detect.py——source human.jpg——save-txt">

——save-txt参数以.txt的形式给出坐标保存到"runs/detect/exp/labels"但是我想把它保存在'runs/data/train'

我认为相关的代码在第108行，但是我不能给出我想要的文件路径。

txt_path = str(save_dir / 'labels' / p.stem) + ('' if dataset.mode == 'image' else f'_{frame}')  # img.txt

我试过了，但是我不能设置名称:

txt_path = 'runs/data/train'  # img.txt

谁知道如何改变txt_path变量可以帮助?

Answer 1

不是一个非常专业的解决方案，但它工作

img_name = p.name.split(".")
txt_path =  f'runs/data/train/{img_name[0]}' # img.txt