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:

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


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

Hi andriizelenko11, this might be useful for your question of converting to TensorRT: GitHub - NVIDIA-AI-IOT/torch2trt: An easy to use PyTorch to TensorRT converter