link之家
链接快照平台
- 输入网页链接，自动生成快照
- 标签化管理网页链接
相关文章推荐
|
失望的鸵鸟 · 陈翔六点半之民间高手：冷檬漂亮，米线高冷，卫 ...· 4 月前 · |
|
|
文质彬彬的仙人掌 · 安徽省药品监督管理局关于亳州市宏宇中药饮片有 ...· 4 月前 · |
|
|
面冷心慈的草稿纸 · 三国志11移动版人物志 ...· 9 月前 · |
|
|
坐怀不乱的小笼包 · 春晚的仪式感，是平凡的温暖· 11 月前 · |
|
|
不拘小节的机器猫 · 凌晨外卖员被邻居嫂子囚禁欺负，三天后他软着腿 ...· 1 年前 · |
# 23.5.9 数据文件清洗。在transform时发现一些cv2.readim()错误的文件，导致训练挂死，需要转换清洗
import os
import cv2
from PIL import Image
# 执行一遍再次执行，可以发现已经没有cv2.imread为空的文件了，清洗完成
directory = "/home/aistudio/work"
for subdir in ["cat_12_predict", "cat_12_train"]:
subdir_path = os.path.join(directory, subdir)
for filename in os.listdir(subdir_path):
filepath = os.path.join(subdir_path, filename)
# 23.5.9 遇到可能的cv2.imread问题的图像文件，先进行转换，避免transform的时候出错
if cv2.imread(filepath) is None:
print(filepath)
img = Image.open(filepath).convert('RGB')
# 23.5.9 先完全遍历并变换格式，让cv2可以识别
img.save(filepath)
print('finish saving')