pytorch模型转换问题

显示全部楼层 · 发表于 2020-8-21 11:32:54

wujialiang 发表于 2020-2-14 17:17
什么时候rknnAPI可以在rk1808上支持c++直接使用pytorch的模型？目前我使用的detectron2训练的mask rcnn ...

请问你转mask rcnn有转成功吗我用pytorch自带的mask rcnn都没转成功

显示全部楼层 · 发表于 2020-8-21 11:46:33

jefferyzhang 发表于 2020-2-16 19:42
rknn-toolkit存在的目的就是倒入其他格式模型，转换成rknn。
rknn-toolkit pc版和arm版功能几乎是一样的 ...

有在pytorch1.2中将torchvision.models.detection.maskrcnn_resnet50_fpn的maskrcnn模型转换示例吗