Dryhten 2021-05-06 01:43 采纳率: 0%
浏览 63

卷积神经网络划分三个数据集内标签文件夹的时候代码有疑问

import os
import config
import shutil
for split in (config.TRAIN,config.VALID,config.TEST):
    print('[INFO] processing {} split:'.format(split))
    imagePaths=os.listdir(os.path.join(config.ORIG_DATA_PATH,split))
    for ele in imagePaths:
        if not ele.endswith('.jpg'):
            imagePaths.remove(ele)
    for imagePath in imagePaths:
        label=config.CLASSES[int(imagePath.split('_')[0])]
        dst=os.path.join(config.BASE_PATH,split,label)
        if not os.path.exists(dst):
            os.makedirs(dst)
        #复制图片
        shutil.copy2(os.path.join(config.ORIG_DATA_PATH,config.TRAIN,imagePath), os.path.join(dst,imagePath))
print('[INFO] All is done' )

这是卷积神经网络的训练集测试集验证集按类别进行划分,我觉得倒数第二行的代码中的config.TRAIN应该改为split,不知道对不对,请解答一下

  • 写回答

4条回答 默认 最新

  • CSDN专家-黄老师 2021-05-06 09:44
    关注

    输出倒数第二行的代码,看看和分析一下数据变化

    评论

报告相同问题?

悬赏问题

  • ¥15 python的qt5界面
  • ¥15 无线电能传输系统MATLAB仿真问题
  • ¥50 如何用脚本实现输入法的热键设置
  • ¥20 我想使用一些网络协议或者部分协议也行,主要想实现类似于traceroute的一定步长内的路由拓扑功能
  • ¥30 深度学习,前后端连接
  • ¥15 孟德尔随机化结果不一致
  • ¥15 apm2.8飞控罗盘bad health,加速度计校准失败
  • ¥15 求解O-S方程的特征值问题给出边界层布拉休斯平行流的中性曲线
  • ¥15 谁有desed数据集呀
  • ¥20 手写数字识别运行c仿真时,程序报错错误代码sim211-100