筛选问题图片数据

最新推荐文章于 2022-06-06 16:10:10 发布

卿勿问

最新推荐文章于 2022-06-06 16:10:10 发布

阅读量110

点赞数

本文链接：https://blog.csdn.net/wilbur520/article/details/118581124

版权

skimage io.imread 图片读取异常处理文件路径

关键词由CSDN通过智能技术生成

import os
# from keras.preprocessing import image
from skimage import io

# imgDir = "/home/hs/tools/rotation/YOLOv5_DOTA_OBB-master/dataliuqunwei/imgs"
imgDir = "/home/hs/tools/rotation/images"
txtDir = "/home/hs/tools/rotation/YOLOv5_DOTA_OBB-master/dataliuqunwei/txts"
xmlsList = os.listdir(imgDir)

for imgfile in xmlsList:
    absolute_path = os.path.join(imgDir, imgfile)
    try:
        io.imread(absolute_path)
    except Exception as e:
        
        # os.remove(absolute_path)
        txtname = imgfile.replace('jpg','txt')
        # os.remove(os.path.join(txtDir, txtname))
        print(absolute_path)
        print(os.path.join(txtDir, txtname))
        continue
    # try:
    #     img= np.array(img, dtype=np.float32)
    # except :
    #     print('corrupt img',absolute_path)
    # EBB6UN7MHFHDVCFJKPTCK4H