Skip to content

This is project about Mask R-CNN using Pytorch and trained on COCO dataset

Notifications You must be signed in to change notification settings

HODUCVU/Mask_R-CNN_with_COCO_dataset

Repository files navigation

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

About

This is project about Mask R-CNN using Pytorch and trained on COCO dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published