skeleton or stgcn module
Opened this issue · 2 comments
hi~how long will the skeleton or st-gcn module be added?
will it take me long time to write a module supporting skeleton and st-gcn？I haven't written a program about tensorrt. :)
Hi
Did you means mmskeleton?
This repo mainly focus on mmdetection, I have no plan to add other open-mmlab codebase convertor here. Sorry.
As for skeleton model, torch2trt_dynamic might meet your demond. It is also my project. Just have a try. If the conversion failed. Let me know , I will see if I can do something.
Add some TensorRT op is not that difficult. For example sum.py. Or write a custom plugin such as repeat, create_repeat_plugin.py, repeatDimsPlugin. PR is welcome.
Yes.
I will try to implement the project about mmskeleton. If there is any progress, I will let you know. Thank you:)