Skip to content

Latest commit

 

History

History
26 lines (19 loc) · 1.33 KB

README.md

File metadata and controls

26 lines (19 loc) · 1.33 KB

Pytorch Mask R-CNN with COCO dataset

Result

Detect and classification on Image

Detect and classification on Video

output_video.mp4

COCO Dataset

Category Link
COCO Dataset saved on Google Drive https://drive.google.com/drive/folders/17pBgoIodgAu9UFRr3NSA_Zul81UOWA1I?usp=sharing
Full dataset from kaggle https://www.kaggle.com/code/karthikeyanvijayan/coco-object-detection-dataset-in-tfrecord/input

Colab

File Link
Train model Training-Pytorch-mask-RCNN-with-COCO-Dataset
Demo on Image Demo-Pytorch-MaskRCNN-on-Image
Demo on Video Demo-Pytorch-MaskRCNN-on-Video
Model Evaluation IoU