Yolo 训练 Error in load_data_detection() - OpenCV 解决办法之一

最新推荐文章于 2024-12-19 10:12:13 发布

alan_山

最新推荐文章于 2024-12-19 10:12:13 发布

阅读量8.2k

点赞数 4

CC 4.0 BY-SA版权

分类专栏：小技巧 python3 文章标签：图像识别

本文链接：https://blog.youkuaiyun.com/u010320736/article/details/110161862

小技巧同时被 2 个专栏收录

9 篇文章

订阅专栏

python3

6 篇文章

订阅专栏

本文介绍了解决YoloV4模型无法识别位深度为8的图片的问题，通过将图片位深度转换为24的方法，成功实现了模型的正常训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近使用YoloV4对验证码模型训练，发现图片位深度为8 的图片无法进行识别，报了个Cannot Load image， Error in load_data_detection() - OpenCV

，

本以为是图片路径问题，结果测试过后，路径没问题，后发现图片的深度是8，将图片位深度转换为24后就可以正常训练了，记录一下，以下是转换代码：

import numpy as np
from PIL import Image
import os

path='img_8/'
newpath='change_24/'
def turnto24(path):
    files = os.listdir(path)
    files = np.sort(files)
    i = 0
    for f in files:
        imgpath = path + f
        img = Image.open(imgpath).convert('RGB')
        dirpath = newpath
        file_name, file_extend = os.path.splitext(f)
        dst = os.path.join(os.path.abspath(dirpath), file_name + '.jpg')
        img.save(dst)

turnto24(path)