yolov5 deepsort训练图片数据集，如何更改代码，使result.txt与每张图片对应

Question

yolov5 deepsort训练图片数据集，如何更改代码，使result.txt，每张图片生成一个txt文件

Answer 1

要实现每张图片生成一个txt文件，您需要更改yolov5_deepsort/utils/results.py中的代码。

打开yolov5_deepsort/utils/results.py文件。
找到def write_results函数。
在函数中找到以下代码块：

for *xyxy, conf, cls in reversed(det):
    if save_txt:  # Write to file
        xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4))).view(-1).tolist()  # normalized xywh
        line = (cls, *xywh, conf) if save_conf else (cls, *xywh)  # label format
        with open(save_path + '.txt', 'a') as f:
            f.write(('%g ' * len(line)).rstrip() % line + '\n')
    if save_img or view_img:  # Add bbox to image
        label = f'{names[int(cls)]} {conf:.2f}'
        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

将上述代码块替换为以下代码块：

for *xyxy, conf, cls in reversed(det):
    if save_txt:  # Write to file
        xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4))).view(-1).tolist()  # normalized xywh
        line = (cls, *xywh, conf) if save_conf else (cls, *xywh)  # label format
        save_file = save_path + f'_{count}.txt'  # Generate unique file name for each image
        with open(save_file, 'w') as f:
            f.write(('%g ' * len(line)).rstrip() % line + '\n')
        count += 1
    if save_img or view_img:  # Add bbox to image
        label = f'{names[int(cls)]} {conf:.2f}'
        plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)

保存文件。
现在，每张图片将生成一个独立的txt文件，文件名为原始图片文件名加上一个数字。例如，如果原始图片文件名为image.jpg，那么生成的txt文件名将为image_0.txt，image_1.txt，依此类推。