(一)使用pytorch完成图片分类任务

参考:http://spytensor.com/index.php/archives/21/?ksxibs=60ei9

pytorch图片分类任务可以分为七部分:数据加载,模型定义,评测标准定义,训练过程定义,验证过程定义,测试过程定义,参数定义 

文件组织结构:

  • dataset/

           --aug.py

           --dataloder.py

  • models/

           --model.py

  • utils/

           --progress_bar.py

           --utils.py

  • checkpoint/

           bestmodels/

  • logs/
  • submits/
  • config.py
  • main.py
  • test.py

  • dataset/:包含两个文件aug.py和dataloder.py。aug.py用于数据增强,dataloader.py用于数据加载。
  • models/:存放一些自定义的模型,如果不使用pytorch自定义的网络模型,可以在这里添加(要添加__init__.py)

           --model.py:定义模型加载

  • utils/:定义常用的评测指标

           --progress_bar.py:进度条输出工具

           --utils.py:定义评测标准,比如accuracy,loss等

  • checkpoint/:存放训练保存的模型

           bestmodels/:保存在验证集上效果最好的模型

  • logs/:存放训练日志
  • config.py:参数定义文件,以参数类的形式定义所需提前设定或修改的参数,例如:数据路径,学习率,训练epoch
  • main.py:主文件,包含训练,测试,验证等过程
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值