search

lxztju / pytorch_classification

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

使用dataloader时加载数据集时出现错误 #5

Closed BarryAlanPan closed 4 years ago

BarryAlanPan commented 4 years ago
image

主要错误集中在“AttributeError: Can't pickle local object 'get-random-eraser..eraser'”

lxztju commented 4 years ago

我刚测试过没有问题，你可以自己调试一下看看哪个地方出错了，如果是get_random_eraser图像增广函数出错了，你可以把它注释掉，效果不会有很大的影响

BarryAlanPan commented 4 years ago

我在注释掉增广函数后，增加一行转换解决了后续的输出问题：

image
QISIRUI commented 4 years ago

我在注释掉增广函数后，增加一行转换解决了后续的输出问题：

image

我也遇到同样的问题，请问是注释掉哪里呢

sky-fly97 commented 4 years ago

+1，我也遇到一样的问题

sky-fly97 commented 4 years ago

我在注释掉增广函数后，增加一行转换解决了后续的输出问题：

image

棒！ 解决了

QISIRUI commented 4 years ago

+1，我也遇到一样的问题

你是怎么解决的啊

sky-fly97 commented 4 years ago

+1，我也遇到一样的问题

你是怎么解决的啊 dataset.py注释掉即可

` 

    self.transform= get_train_transform(size=cfg.INPUT_SIZE)
    # self.eraser = get_random_eraser( s_h=0.1, pixel_level=True)
    self.input_size = cfg.INPUT_SIZE`
QISIRUI commented 4 years ago

然而，我还是出现另一个问题@sky-fly97 pybug1

也不知道是不是和数据集格式有关

thevolga commented 3 years ago

然而，我还是出现另一个问题@sky-fly97 pybug1

也不知道是不是和数据集格式有关

同出现这个问题 请问解决了吗