是不是少了一些文件呢

lxztju / pytorch_classification

利用pytorch实现图像分类的一个完整的代码，训练，预测，TTA，模型融合，模型部署，cnn提取特征，svm或者随机森林等进行分类，模型蒸馏，一个完整的代码

MIT License

1.38k stars 339 forks source link

Open chuanfuye opened 2 years ago

chuanfuye commented 2 years ago

在data文件夹里面 from . import random_eraser 没有找到random_eraser模块

VixeruntR commented 2 years ago

作者把这个模块删掉了，你把导入这个的代码都注释掉就行了

Zhang-Jinwei commented 2 years ago

注释掉了也不行吧，后面很多的代码都要用到他导入的train_datasets