Skip to content
use pytorch to do image classfiication tasks
Python
Branch: master
Clone or download
Latest commit 516a6a7 Oct 26, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
dataset add albumentations Oct 7, 2019
models update Dec 19, 2018
utils update to version0.2.0 Dec 19, 2018
README.md Update README.md Jun 23, 2019
config.py add albumentations Oct 7, 2019
main.py Update main.py Oct 26, 2019
test.py split test from main.py to test.py Apr 28, 2019

README.md

pytorch-image-classification

use pytorch to do image classfiication tasks

使用方法

移步:从实例掌握 pytorch 进行图像分类

2019.04.28

add test.py

version0.2.0

更新 手写了一份进度条工具,效果如下

loading train dataset
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4572/4572 [00:00<00:00, 1145746.42it/s]
loading train dataset
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 2520/2520 [00:00<00:00, 1128994.45it/s]
Train Epoch:  1/40 [>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>]  [Current: Loss 1.079473 Top1: 75.131233 ]  286/286 [ 100% ]
Val   Epoch:  1/40 [>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>]  [Current: Loss 0.469368 Top1: 89.484131 ]  79/79 [ 100% ]
Get Better top1 : 89.4841 saving weights to ./checkpoints/best_model/resnet50/0/model_best.pth.tar
Train Epoch:  2/40 [>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>]  [Current: Loss 0.696026 Top1: 83.442696 ]  286/286 [ 100% ]
Val   Epoch:  2/40 [>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>]  [Current: Loss 0.275893 Top1: 91.269844 ]  79/79 [ 100% ]
Get Better top1 : 91.2698 saving weights to ./checkpoints/best_model/resnet50/0/model_best.pth.tar
Train Epoch:  3/40 [>>>>>                                             ]  [Current: Loss 0.667610 Top1: 84.165291 ]  31/286 [  10% ]
You can’t perform that action at this time.