├── .gitignore ├── 2018_AL_defect_train.py ├── 2018_bd_tiangong_train.py ├── 2018_bd_xjtu_train.py ├── 2018_pdr_train.py ├── README.md ├── dataset ├── AL_defect_dataset.py ├── __init__.py ├── bd_tiangong_dataset.py ├── bd_xjtu_dataset.py ├── data_aug.py └── pdr_dataset.py ├── models ├── __init__.py ├── inception_resnet_v2.py ├── inception_v4.py ├── multiscale_resnet.py └── xception.py ├── predict ├── bd_tiangong_model_merge.py ├── bd_tiangong_pred.py ├── bd_tiangong_pred_unknow.py └── bd_xjtu_pred.py └── utils ├── __init__.py ├── losses.py └── train_util.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | *.txt 3 | *.pth 4 | predict/Baidu/ 5 | .idea -------------------------------------------------------------------------------- /2018_AL_defect_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/2018_AL_defect_train.py -------------------------------------------------------------------------------- /2018_bd_tiangong_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/2018_bd_tiangong_train.py -------------------------------------------------------------------------------- /2018_bd_xjtu_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/2018_bd_xjtu_train.py -------------------------------------------------------------------------------- /2018_pdr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/2018_pdr_train.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/README.md -------------------------------------------------------------------------------- /dataset/AL_defect_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/dataset/AL_defect_dataset.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/bd_tiangong_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/dataset/bd_tiangong_dataset.py -------------------------------------------------------------------------------- /dataset/bd_xjtu_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/dataset/bd_xjtu_dataset.py -------------------------------------------------------------------------------- /dataset/data_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/dataset/data_aug.py -------------------------------------------------------------------------------- /dataset/pdr_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/dataset/pdr_dataset.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/models/inception_resnet_v2.py -------------------------------------------------------------------------------- /models/inception_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/models/inception_v4.py -------------------------------------------------------------------------------- /models/multiscale_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/models/multiscale_resnet.py -------------------------------------------------------------------------------- /models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/models/xception.py -------------------------------------------------------------------------------- /predict/bd_tiangong_model_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/predict/bd_tiangong_model_merge.py -------------------------------------------------------------------------------- /predict/bd_tiangong_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/predict/bd_tiangong_pred.py -------------------------------------------------------------------------------- /predict/bd_tiangong_pred_unknow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/predict/bd_tiangong_pred_unknow.py -------------------------------------------------------------------------------- /predict/bd_xjtu_pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/predict/bd_xjtu_pred.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OdingdongO/pytorch_classification/HEAD/utils/train_util.py --------------------------------------------------------------------------------