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

使用自定义的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

posted @ 2020-03-22 23:33  vivia~  阅读(1285)  评论(0编辑  收藏  举报