Skip to content

wmathor/pytorch-garbage-classification

Repository files navigation

garbage-classification

垃圾分类-数据分析和预处理

  • 整体数据探测
  • 分析数据不同类别分布
  • 分析图片长宽比例分布
  • 切分数据集和验证集
  • 数据可视化展示(可视化工具 pyecharts,seaborn,matplotlib)

代码结构

├── app_WSL-Images_resnext.py
├── app_garbage.py
├── args.py
├── checkpoint
│   ├── checkpoint.pth.tar
│   ├── garbage_resnext101_model_9_9547_9588.pth
│   └── resnext101_16d_log_iter10_cuda.txt
├── data
│   ├── ImageNet1k_label.txt
│   ├── class.txt
│   ├── garbage-classify-for-pytorch
│   ├── garbage_label.txt
│   ├── train1.txt
│   └── val1.txt
|
├── garbage-classification-using-pytorch.py
├── json_utils.py
├── model.py
├── models
│   ├── alexnet.py
│   ├── densenet.py
│   ├── inception.py
│   ├── resnet.py
│   ├── squeezenet.py
│   └── vgg.py
├── preprocess
├── transform.py
├── utils
│   ├── eval.py
│   ├── json_utils.py
│   ├── logger.py
│   └── misc.py

resnext101网络架构

  • pre_trained_model resnext101 网络架构原理
  • 基于pytorch 数据处理、resnext101 模型分类预测
  • 在线服务API 接口

垃圾分类-训练

python garbage-classification-using-pytorch.py
--model_name resnext101_32x16d
--lr 0.001
--optimizer adam
--start_epoch 1
--epochs 2
--num_classes 4

  • model_name 模型名称
  • lr 学习率
  • optimizer 优化器
  • start_epoch 训练过程断点重新训练
  • num_classes 分类个数

垃圾分类-评估

python garbage-classification-using-pytorch.py
--model_name resnext101_32x16d
--evaluate
--resume checkpoint/checkpoint.pth.tar
--num_classes 4

  • model_name 模型名称
  • evaluate 模型评估
  • resume 指定checkpoint 文件路径,保存模型以及训练过程参数

垃圾分类-在线预测

python app_garbage.py
--model_name resnext101_32x16d
--resume checkpoint/garbage_resnext101_model_9_9547_9588.pth

  • model_name 模型名称
  • resume 训练模型文件路径
  • 模型预测
    命令行验证和postman 方式验证
    举例说明:命令行模式下预测
    curl -X POST -F file=@cat.jpg http://ip:port/predict

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published