Skip to content

Colab-friendly implementation of MaskRCNN in PyTorch with ResNet18 and ResNet50 backends. Fit for image classification, object detection, and segmentation.

Notifications You must be signed in to change notification settings

atherfawaz/Mask-RCNN-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image classification and Object detection + segmentation on a custom dataset of ID cards

Colab-friendly project written in Pytorch.

  • Image classification using Resnet18.
  • Object detection and segmentation using Mask RCNN with Resnet50 backbone.

You can view the Jupyter Notebook here.