facebookresearch / Mask2Former

Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"
MIT License
2.59k stars 388 forks source link

Demo prediction gets killed in Jetson Nano #194

Open kanishkanarch opened 1 year ago

kanishkanarch commented 1 year ago

Hi, I am trying the COCO panoptic model with ResNet50 backbone on this image on a Jetson Nano having JetPack 4.6.3 and CUDA 10.2. image The demo gets killed each time but maybe not because of some memory issue, here are my memory stats. Please tell me if any other information is required.