ultralytics yolo系列 detect无法正确保存中文路径

最新推荐文章于 2024-05-05 17:35:02 发布

K. O. Long

最新推荐文章于 2024-05-05 17:35:02 发布

阅读量1k

点赞数

分类专栏： yolov5 文章标签： python opencv 开发语言

本文链接：https://blog.csdn.net/weixin_44669218/article/details/128007613

版权

yolov5 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

这篇博客主要讨论了在Python代码中遇到的中文路径保存导致的乱码问题。作者指出了问题发生的两个关键位置：detect.py中的cv2.imwrite函数和utils/plot.py中的save_one_box函数。为了解决这个问题，作者建议将cv2.imwrite替换为cv2.imencode，并将编码后的数据写入文件。这种方法成功解决了中文路径保存时的乱码问题。

摘要由CSDN通过智能技术生成

问题源头

detect.py 行 186 左右位置

            # Save results (image with detections)
            if save_img:
                if dataset.mode == 'image':
                    cv2.imencode('.jpg', im0)[1].tofile(save_path)
                    # cv2.imwrite(save_path, im0)

如果打开了save_crop选项，则还需修改utils/plot.py 行 458 左右位置
（这部分问题仅限老版本，新版本使用的PIL Image保存不知道会不会有这个错误）

def save_one_box(xyxy, im, file='image.jpg', gain=1.02, pad=10, square=False, BGR=False, save=True):
    # Save image crop as {file} with crop size multiple {gain} and {pad} pixels. Save and/or return crop
    xyxy = torch.tensor(xyxy).view(-1, 4)
    b = xyxy2xywh(xyxy)  # boxes
    if square:
        b[:, 2:] = b[:, 2:].max(1)[0].unsqueeze(1)  # attempt rectangle to square
    b[:, 2:] = b[:, 2:] * gain + pad  # box wh * gain + pad
    xyxy = xywh2xyxy(b).long()
    clip_coords(xyxy, im.shape)
    crop = im[int(xyxy[0, 1]):int(xyxy[0, 3]), int(xyxy[0, 0]):int(xyxy[0, 2]), ::(1 if BGR else -1)]
    if save:
        file.parent.mkdir(parents=True, exist_ok=True)  # make directory
        # 下一行的imencode是更改后的
        cv2.imencode('.jpg', crop)[1].tofile(str(increment_path(file).with_suffix('.jpg')))
    return crop