pytorch_classification icon indicating copy to clipboard operation
pytorch_classification copied to clipboard

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

Results 42 pytorch_classification issues
Sort by recently updated
recently updated
newest added

你好,想问一下,出现ModuleNotFoundError: No module named 'cfg'需要怎么解决呀?谢谢

是什么的特征值?怎么求到这个2816的?

我使用resnet101网络模型,train和predict都可以正常运行,但运行cnn_ml出错了,大佬可以解答下吗?

RuntimeError: “nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘

![1648558977(1)](https://user-images.githubusercontent.com/86580598/160617304-75177bce-c4ef-47cf-bc67-67fe54b4cf47.png) 这一块数据集导入,是不是有点问题呀,index是什么东西呀,label也读不出来。一直显示下面的错误 ![image](https://user-images.githubusercontent.com/86580598/160618461-9a97c470-e33a-4807-8d8f-919fd2399bb0.png)

当自己尝试使用resnet50或者moblienetv2时都会报如下错误: `Traceback (most recent call last): File "train.py", line 147, in images, labels = next(batch_iterator) File "/home/xuhaowang/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 363, in __next__ data = self._next_data() File "/home/xuhaowang/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 403, in _next_data...

请教大佬这个bug应该怎么解决呢 ![pybug2](https://user-images.githubusercontent.com/48178028/88806987-b86aa680-d1e3-11ea-9c7e-d1dff9b7c41e.png) 另外,在train_kd.py中,定义优化器与损失函数那里,model是标红的,直接运行,显示没有定义; 当用teacher_model,student_model代替model时,显示下面这个bug。 optimizer = optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=cfg.LR) optimizer = optim.SGD(model.parameters(), lr=cfg.LR, momentum=cfg.MOMENTUM, weight_decay=cfg.WEIGHT_DECAY) ![pybug3](https://user-images.githubusercontent.com/48178028/88807928-d389e600-d1e4-11ea-9e90-b51c351533d2.png) 抱拳

cnn_svm中的 get_test_transform 并未定义,是不是写错了?