classification-pytorch icon indicating copy to clipboard operation
classification-pytorch copied to clipboard

这是各个主干网络分类模型的源码,可以用于训练自己的分类模型。

Results 27 classification-pytorch issues
Sort by recently updated
recently updated
newest added

![0d241d490ba5eeba2bcdb04212cd88b](https://user-images.githubusercontent.com/61875144/170970111-58dc79d3-27c5-46b6-984d-bc3c6fa3ec82.png) 泡导,我没有改过您的代码,运行的时候却出现了这个问题,导致没法完成训练,求指教。 显卡:3050ti 环境:全部按您的教程配置的pytorch1.71、torchvision0.8.2

学习率一直是0.0001,不会变动,这个有办法解决嘛

SSD网络改了后面的网络,训练了100epochs,map只有61.9%正常吗,训练前用原始权重有79.3%

出错信息: ``` There is no pretrained model for vit_b_16 Traceback (most recent call last): File "train.py", line 225, in model = get_model_from_name[backbone](input_shape = input_shape, num_classes = num_classes, pretrained = pretrained)...

百度云盘没会员太慢了,求个Google Drive

为什么验证精度比训练精度高?高的太离谱了 ![image](https://github.com/bubbliiiing/classification-pytorch/assets/88662555/aada5deb-c8ba-4d69-8d17-1153dbb4963d)

代码段如下: if not random: image = self.resize(image) image = self.center_crop(image) return image 请问验证集图片预处理的目的是什么,验证集的结果是否影响了训练集的超参数更新? 谢谢!