Skip to content

xalphonseinbaraj/grmrcnn

Repository files navigation

This python package is used for object detection that combines Gradient-weighted Class Activation Mapping++ (Grad-CAM++) for localization and Mask Regional Convolution Neural Network (Mask R-CNN) for object detection. Hence this model is knowns as GradCAM-MLRCNN

image This is architecture of the model

We are applied this model for various pretrained models like VGG16,VGG19,ResNet 101, ResNet 152 and ResNet 50. This python package is exclusively for ResNet 50 only. For remaining pretrained models, still under development process

The main process of this models as follows

  1. It will predict the Class Activation Map (CAM) on given input image image

  2. For localization, we used Gradient-Weighted Class Activation Map (Grad-CAM++) and for object detection, we used Mask Regional Convolutinal Neural Network (Mask R-CNN)

image

Use the following code:

                                    from tensorflow import keras
                                    import glob
                                    from grmrcnn import grmrcnn
                                    #need to mention and keep the model in the current directory (ResNet -Recommended)
                                    model = keras.models.load_model("ResNet50.h5") 
                                    path = glob.glob("...../Images/*.jpg") #Image Path
                                    GradCAMMRCNN=grmrcnn(model,path)

Installation

Requirements: Python 3.7

Note: We used tensorflow 1.x version and keras 2.3.1. Finally, install this module from pypi using pip.

                                              pip install grmrcnn

How this model(Grad-CAM++ with Mask RCNN) works:

If you want to know more about GradCAM++ with Mask RCNN, read our article

Thanks

Many, many thanks to our advisor professor Dr.Jyh-Horng Jeng and co-advisor Dr.Jer-Guang Hsieh for guiding and helping for this amazing model and our collegues Mr.Julio Jerison Macrohon and Ms.Charlyn Villavicencio to publish this article successfully.

This research was funded as a scholar of the Ministry of Science and Technology (MOST), Taiwan and I-Shou University, Kaohsiung City, Taiwan.

Thanks to everyone who works on all the awesome Python data science libraries like numpy, scipy, scikit-image, pillow, etc, etc that makes this kind of stuff so easy and fun in Python.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages