https://www.lmlphp.com/user/62501/article/item/1225624/

train.py

网络训练主函数,主要操作有:

  1. 传入训练参数;通常采用argparse库,支持脚本传入。
  2. 网络训练;包括定义网络、加载模型、前向反向传播、保存模型等。
  3. 将训练情况可视化;使用tensorboard绘制loss曲线。

datasets.py

在pytorch中数据加载到模型的操作顺序如下:

  1. 创建一个Dataset对象,一般重载__len____getitem__方法。__len__返回数据集大小,__getitem__支持索引,以便Dataset[i]获取第i个样本。
  2. 创建一个DataLoader对象,将Dataset作为参数传入。
  3. 循环这个DataLoader对象,将img、label加载到模型中进行训练。

我们还需在Dataset对象中定义数据预处理,这里采用:

  1. 0.7-1.4倍的随机尺度缩放

  2. 各通道减去ImageNet的均值

  3. 随机crop下769x769大小

  4. 镜像随机翻转

注意:为了让Image和Label对应,也要对Label作相应的预处理,具体过程详见代码。

最新文章

  1. telnet测试制定地址端口号
  2. NodeJS优缺点及适用场景讨论
  3. poj 2446 Chessboard (二分匹配)
  4. winform 表单正则表达式验证 示例(ValidationRule)
  5. jQuery各种效果举例
  6. js 全国城市3级联动
  7. JavaScript导航树
  8. leetcode 224. Basic Calculator 、227. Basic Calculator II
  9. MVP 实践
  10. C# 6.0:在catch和finally中使用await
  11. Win2012R2 AD主域控登录密码忘记
  12. linux 安装mysql相关和openjdk
  13. kettle的资源库创建及修改密码
  14. MUI - IOS系统,相册选择照片后,点击确定按钮无反应
  15. WPF自定义控件的自定义属性绑定后不更新问题
  16. sublime text 3 使用简介
  17. 如何将Jenkins multiline string parameter的多行文本优雅的保存为文件
  18. [cogs347]地震
  19. 以二进制方式读取图片保存到string
  20. IOS开发 多线程编程 - NSOperationQueue

热门文章

  1. journalctl查看内核/应用日志
  2. RabbitMQ管理界面使用之手动送数据
  3. go 语言的宕机恢复(recover)
  4. Java线程安全之synchronized 与 lock的异同
  5. Java基础__05.网络编程
  6. List<Object>转为对象
  7. MSDTC突然停用了,后台数据无法更新
  8. Ingress限流
  9. 个人常用的win7快捷键
  10. ConstantBuffer