首页 > 其他 > 详细

使用自定义的Dataloader做数据增强、格式统一等操作/像使用pytorch一样进行训练。

时间:2020-03-23 01:40:35      阅读:692      评论:0      收藏:0      [点我收藏+]

使用自定义的Dataloader做数据增强、格式统一等操作/像使用pytorch一样进行训练。

格式统一

https://detectron2.readthedocs.io/tutorials/data_loading.html

不使用train而是使用Model进行自定义训练

https://detectron2.readthedocs.io/tutorials/models.html

实现并写一个新的model层,注册到config以供使用

https://detectron2.readthedocs.io/tutorials/write-models.html

自定义整个训练过程的方式

https://github.com/facebookresearch/detectron2/blob/master/tools/plain_train_net.py

使用自定义的Dataloader做数据增强、格式统一等操作/像使用pytorch一样进行训练。

原文:https://www.cnblogs.com/miaozhijuan/p/12549400.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!