码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-03-23 00:17:46      阅读:107      评论:0      收藏:0      [点我收藏+]

标签:block   ref   width   htm   down   any   too   render   rem   

使用自定义的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一样进行训练。

标签:block   ref   width   htm   down   any   too   render   rem   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!