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

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