实例介绍
pytorch_classification 利用pytorch实现图像分类,其中包含的密集网,resnext,mobilenet,efficiencynet,resnet等图像分类网络,可以根据需要再行利用torchvision扩展其他的分类算法 实现功能 基础功能利用pytorch实现图像分类 包含带有warmup的cosine学习率调整 warmup的step学习率优调整 多模型融合预测,修正与投票融合 利用flask实现模型云端api部署 使用tta测试时增强进行预测 添加label smooth的pytorch实现(标签平滑) 添加使用cnn提取特征,并使用SVM,RF,MLP,KN
【实例截图】
【核心代码】
16359647549157788269.zip
└── pytorch_classification-master
├── cfg.py
├── cnn_ml.py
├── data
│ ├── dataset.py
│ ├── __init__.py
│ ├── preprocess.py
│ ├── __pycache__
│ │ ├── dataset.cpython-37.pyc
│ │ ├── __init__.cpython-37.pyc
│ │ ├── preprocess.cpython-37.pyc
│ │ ├── random_eraser.cpython-37.pyc
│ │ └── transform.cpython-37.pyc
│ ├── random_eraser.py
│ └── transform.py
├── deployment
│ ├── c++_libtorch
│ │ ├── CMakeLists.txt
│ │ ├── main.cpp
│ │ └── SaveTorchscriptModel.py
│ ├── flask_v1
│ │ ├── client.py
│ │ ├── server.py
│ │ └── transform.py
│ ├── flask_v2
│ │ ├── cfg.py
│ │ ├── models
│ │ │ ├── build_model.py
│ │ │ ├── efficientnet_pytorch
│ │ │ │ ├── __init__.py
│ │ │ │ ├── model.py
│ │ │ │ ├── __pycache__
│ │ │ │ │ ├── __init__.cpython-36.pyc
│ │ │ │ │ ├── __init__.cpython-37.pyc
│ │ │ │ │ ├── model.cpython-36.pyc
│ │ │ │ │ ├── model.cpython-37.pyc
│ │ │ │ │ ├── utils.cpython-36.pyc
│ │ │ │ │ └── utils.cpython-37.pyc
│ │ │ │ └── utils.py
│ │ │ ├── __init__.py
│ │ │ ├── __pycache__
│ │ │ │ ├── build_model.cpython-36.pyc
│ │ │ │ ├── build_model.cpython-37.pyc
│ │ │ │ ├── __init__.cpython-36.pyc
│ │ │ │ ├── __init__.cpython-37.pyc
│ │ │ │ ├── resnext_wsl.cpython-36.pyc
│ │ │ │ └── resnext_wsl.cpython-37.pyc
│ │ │ ├── resnext_wsl.py
│ │ │ └── vision
│ │ │ ├── alexnet.py
│ │ │ ├── densenet.py
│ │ │ ├── detection
│ │ │ │ ├── backbone_utils.py
│ │ │ │ ├── faster_rcnn.py
│ │ │ │ ├── generalized_rcnn.py
│ │ │ │ ├── image_list.py
│ │ │ │ ├── __init__.py
│ │ │ │ ├── keypoint_rcnn.py
│ │ │ │ ├── mask_rcnn.py
│ │ │ │ ├── __pycache__
│ │ │ │ │ ├── backbone_utils.cpython-36.pyc
│ │ │ │ │ ├── backbone_utils.cpython-37.pyc
│ │ │ │ │ ├── faster_rcnn.cpython-36.pyc
│ │ │ │ │ ├── faster_rcnn.cpython-37.pyc
│ │ │ │ │ ├── generalized_rcnn.cpython-36.pyc
│ │ │ │ │ ├── generalized_rcnn.cpython-37.pyc
│ │ │ │ │ ├── image_list.cpython-36.pyc
│ │ │ │ │ ├── image_list.cpython-37.pyc
│ │ │ │ │ ├── __init__.cpython-36.pyc
│ │ │ │ │ ├── __init__.cpython-37.pyc
│ │ │ │ │ ├── keypoint_rcnn.cpython-36.pyc
│ │ │ │ │ ├── keypoint_rcnn.cpython-37.pyc
│ │ │ │ │ ├── mask_rcnn.cpython-36.pyc
│ │ │ │ │ ├── mask_rcnn.cpython-37.pyc
│ │ │ │ │ ├── roi_heads.cpython-36.pyc
│ │ │ │ │ ├── roi_heads.cpython-37.pyc
│ │ │ │ │ ├── rpn.cpython-36.pyc
│ │ │ │ │ ├── rpn.cpython-37.pyc
│ │ │ │ │ ├── transform.cpython-36.pyc
│ │ │ │ │ ├── transform.cpython-37.pyc
│ │ │ │ │ ├── _utils.cpython-36.pyc
│ │ │ │ │ └── _utils.cpython-37.pyc
│ │ │ │ ├── roi_heads.py
│ │ │ │ ├── rpn.py
│ │ │ │ ├── transform.py
│ │ │ │ └── _utils.py
│ │ │ ├── googlenet.py
│ │ │ ├── inception.py
│ │ │ ├── __init__.py
│ │ │ ├── mobilenet.py
│ │ │ ├── __pycache__
│ │ │ │ ├── alexnet.cpython-36.pyc
│ │ │ │ ├── alexnet.cpython-37.pyc
│ │ │ │ ├── densenet.cpython-36.pyc
│ │ │ │ ├── densenet.cpython-37.pyc
│ │ │ │ ├── googlenet.cpython-36.pyc
│ │ │ │ ├── googlenet.cpython-37.pyc
│ │ │ │ ├── inception.cpython-36.pyc
│ │ │ │ ├── inception.cpython-37.pyc
│ │ │ │ ├── __init__.cpython-36.pyc
│ │ │ │ ├── __init__.cpython-37.pyc
│ │ │ │ ├── mobilenet.cpython-36.pyc
│ │ │ │ ├── mobilenet.cpython-37.pyc
│ │ │ │ ├── resnet.cpython-36.pyc
│ │ │ │ ├── resnet.cpython-37.pyc
│ │ │ │ ├── shufflenetv2.cpython-36.pyc
│ │ │ │ ├── shufflenetv2.cpython-37.pyc
│ │ │ │ ├── squeezenet.cpython-36.pyc
│ │ │ │ ├── squeezenet.cpython-37.pyc
│ │ │ │ ├── _utils.cpython-36.pyc
│ │ │ │ ├── utils.cpython-36.pyc
│ │ │ │ ├── _utils.cpython-37.pyc
│ │ │ │ ├── utils.cpython-37.pyc
│ │ │ │ ├── vgg.cpython-36.pyc
│ │ │ │ └── vgg.cpython-37.pyc
│ │ │ ├── resnet.py
│ │ │ ├── segmentation
│ │ │ │ ├── deeplabv3.py
│ │ │ │ ├── fcn.py
│ │ │ │ ├── __init__.py
│ │ │ │ ├── __pycache__
│ │ │ │ │ ├── deeplabv3.cpython-36.pyc
│ │ │ │ │ ├── deeplabv3.cpython-37.pyc
│ │ │ │ │ ├── fcn.cpython-36.pyc
│ │ │ │ │ ├── fcn.cpython-37.pyc
│ │ │ │ │ ├── __init__.cpython-36.pyc
│ │ │ │ │ ├── __init__.cpython-37.pyc
│ │ │ │ │ ├── segmentation.cpython-36.pyc
│ │ │ │ │ ├── segmentation.cpython-37.pyc
│ │ │ │ │ ├── _utils.cpython-36.pyc
│ │ │ │ │ └── _utils.cpython-37.pyc
│ │ │ │ ├── segmentation.py
│ │ │ │ └── _utils.py
│ │ │ ├── shufflenetv2.py
│ │ │ ├── squeezenet.py
│ │ │ ├── _utils.py
│ │ │ ├── utils.py
│ │ │ └── vgg.py
│ │ ├── redis_db.py
│ │ ├── server.py
│ │ ├── streess_test.py
│ │ └── utils.py
│ └── README.md
├── ensamble
│ ├── kaggle_vote.py
│ └── samples
│ ├── method_1.csv
│ ├── method_2.csv
│ └── method_3.csv
├── LICENSE
├── models
│ ├── build_model.py
│ ├── efficientnet_pytorch
│ │ ├── __init__.py
│ │ ├── model.py
│ │ ├── __pycache__
│ │ │ ├── __init__.cpython-37.pyc
│ │ │ ├── model.cpython-37.pyc
│ │ │ └── utils.cpython-37.pyc
│ │ └── utils.py
│ ├── __init__.py
│ ├── __pycache__
│ │ ├── build_model.cpython-37.pyc
│ │ ├── __init__.cpython-37.pyc
│ │ └── resnext_wsl.cpython-37.pyc
│ ├── resnext_wsl.py
│ └── vision
│ ├── alexnet.py
│ ├── densenet.py
│ ├── detection
│ │ ├── backbone_utils.py
│ │ ├── faster_rcnn.py
│ │ ├── generalized_rcnn.py
│ │ ├── image_list.py
│ │ ├── __init__.py
│ │ ├── keypoint_rcnn.py
│ │ ├── mask_rcnn.py
│ │ ├── __pycache__
│ │ │ ├── backbone_utils.cpython-37.pyc
│ │ │ ├── faster_rcnn.cpython-37.pyc
│ │ │ ├── generalized_rcnn.cpython-37.pyc
│ │ │ ├── image_list.cpython-37.pyc
│ │ │ ├── __init__.cpython-37.pyc
│ │ │ ├── keypoint_rcnn.cpython-37.pyc
│ │ │ ├── mask_rcnn.cpython-37.pyc
│ │ │ ├── roi_heads.cpython-37.pyc
│ │ │ ├── rpn.cpython-37.pyc
│ │ │ ├── transform.cpython-37.pyc
│ │ │ └── _utils.cpython-37.pyc
│ │ ├── roi_heads.py
│ │ ├── rpn.py
│ │ ├── transform.py
│ │ └── _utils.py
│ ├── googlenet.py
│ ├── inception.py
│ ├── __init__.py
│ ├── mobilenet.py
│ ├── __pycache__
│ │ ├── alexnet.cpython-37.pyc
│ │ ├── densenet.cpython-37.pyc
│ │ ├── googlenet.cpython-37.pyc
│ │ ├── inception.cpython-37.pyc
│ │ ├── __init__.cpython-37.pyc
│ │ ├── mobilenet.cpython-37.pyc
│ │ ├── resnet.cpython-37.pyc
│ │ ├── shufflenetv2.cpython-37.pyc
│ │ ├── squeezenet.cpython-37.pyc
│ │ ├── _utils.cpython-37.pyc
│ │ ├── utils.cpython-37.pyc
│ │ └── vgg.cpython-37.pyc
│ ├── resnet.py
│ ├── segmentation
│ │ ├── deeplabv3.py
│ │ ├── fcn.py
│ │ ├── __init__.py
│ │ ├── __pycache__
│ │ │ ├── deeplabv3.cpython-37.pyc
│ │ │ ├── fcn.cpython-37.pyc
│ │ │ ├── __init__.cpython-37.pyc
│ │ │ ├── segmentation.cpython-37.pyc
│ │ │ └── _utils.cpython-37.pyc
│ │ ├── segmentation.py
│ │ └── _utils.py
│ ├── shufflenetv2.py
│ ├── squeezenet.py
│ ├── _utils.py
│ ├── utils.py
│ └── vgg.py
├── predict.py
├── __pycache__
│ └── cfg.cpython-37.pyc
├── README.md
├── train_kd.py
├── train.py
├── train_val.py
├── utils
│ ├── __init__.py
│ ├── label_smoothing_pytorch.py
│ ├── loss_kd.py
│ └── warmup_lr.py
└── Visualization
├── f1_conv1.png
├── Feature_Visualization.ipynb
├── Feature_Visualization.py
└── test.png
32 directories, 213 files
标签:
pytorch_classification:利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型...
小贴士
感谢您为本站写下的评论,您的评论对其它用户来说具有重要的参考价值,所以请认真填写。
- 类似“顶”、“沙发”之类没有营养的文字,对勤劳贡献的楼主来说是令人沮丧的反馈信息。
- 相信您也不想看到一排文字/表情墙,所以请不要反馈意义不大的重复字符,也请尽量不要纯表情的回复。
- 提问之前请再仔细看一遍楼主的说明,或许是您遗漏了。
- 请勿到处挖坑绊人、招贴广告。既占空间让人厌烦,又没人会搭理,于人于己都无利。
关于好例子网
本站旨在为广大IT学习爱好者提供一个非营利性互相学习交流分享平台。本站所有资源都可以被免费获取学习研究。本站资源来自网友分享,对搜索内容的合法性不具有预见性、识别性、控制性,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,平台无法对用户传输的作品、信息、内容的权属或合法性、安全性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论平台是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二与二十三条之规定,若资源存在侵权或相关问题请联系本站客服人员,点此联系我们。关于更多版权及免责申明参见 版权及免责申明
网友评论
我要评论