系列文章地址:深度学习基础网络实现-csdn
框架:pytorch1.8.1
python:python3.7
模型 | LeNet | AlexNet | VggNet | GoogLeNet | ResNet |
---|---|---|---|---|---|
轮数 | 5 | 5 | 5 | 5 | 5 |
精度 | 0.622 | 0.800、0.860(加pretrain) | 0.911(加pretrain) | 0.932(加pretrain) | 0.927(加pretrain) |
日志 | lenet.log | alexnet.log、alexnet_pretrain.log | vggnet.log | googlenet.log | resnet.log |
- model: 选择使用的模型(字符串类型,例如:--model lenet)
- save_feature: 设置是否保存特征图(布尔类型,例如:--save_feature True)
python3 train.py --model lenet
python3 predict.py --model lenet
python3 params_print.py --model lenet