how to export the Export_d2go_trained_coco2017_torchscript_int8_mask_rcnn for custom class??

tsubauaaa / flutter_d2go

Flutter plugin that runs d2go

BSD 3-Clause "New" or "Revised" License

10 stars 10 forks source link

how to export the Export_d2go_trained_coco2017_torchscript_int8_mask_rcnn for custom class?? #12

Open VaibhavAsare opened 1 year ago

VaibhavAsare commented 1 year ago

hello , @tsubauaaa Can you please help me to Export_d2go_trained_coco2017_torchscript_int8_mask_rcnn for CUSTOM CLASSES ? As you have done for the pre trained model in the given notebook and it is working for me as well, But I have to export the custom classes based model into torchscript_int8 using mask RCNN. thanks in advance