Custom Instance Segmentation

Custom instance segmentation using detectron2:
GitHub: Custom-object-detection-using-Detectron2/Custom_Instance_Segmentation.ipynb at main · jakariaemon/Custom-object-detection-using-Detectron2 · GitHub

Custom Dataset: https://app.roboflow.com/ds/iB0WrTySI1?key=YTXbygp65f

Trained on RTX 3060 and Tested on Jetson Nano (4GB).
Installation of Detectron2 is straight forward.

2 Likes

Have you convert torch weights-model into tensorrt or do you know how to do it, because inferencing net from pytorch on jetson means loosing most of computational performance of the device