search

airockchip / rknn-toolkit2

Other
844 stars 88 forks source link

can we deploy Faster RCNN on Rock 3588 NPU? #102

Open apanand14 opened 2 months ago

apanand14 commented 2 months ago

Hello, I would like to run faster rcnn on rockpi5b using NPU? is it possible to convert .pth models to .rknn using toolkit?

Thank you for your answer!!

yuyun2000 commented 2 months ago

Yes, it is very likely to work.

apanand14 commented 2 months ago

thank you @yuyun2000 . is there any demo support like yolo models fro fasterrcnn and how to convert the model?

yuyun2000 commented 2 months ago

https://github.com/airockchip/rknn-toolkit2/tree/master/rknn-toolkit2/examples/onnx/yolov5

apanand14 commented 2 months ago

@yuyun2000 I'm aware of yolo model support for rknn but I'm asking for fastercnn object detection model. is it supported by rknn?

yuyun2000 commented 2 months ago

You need to explore it yourself. In my opinion, he definitely supports it.