Skip to content

pkhigh/ConUHack2020

Repository files navigation

Intact Challenge using Computer Vision : 
    In this challenge, I try to solve the problem of autmoatic car model detection in an image.
    This information can be used to get the car model type of moving vehicles.
    Also, here along with knowing the car type I also localize the car in the scene which can be further used in various
    information extraction like license plate, damage location.
    The computer vision models are trained on 8k images of 196 classes of cars.
    Here I used transfer learning to train Mask-RCNN ResNet-50 model trained on COCO dataset as my initial starting point.
    
STATISTICS:
Dataset : Cars Dataset
Number of Classes : 196
Training framework used : detectron2 (Recently open sourced by Facebook)
Training Images : 8144
Testing Images : 8041

MAP(Mean Average Precision over all the classes) achieved by the best model on :

Training Data : 
AP,AP50,AP75,APs,APm,APl
98.7940,99.9520,99.9520,-100.0000,100.0000,98.7940

Validation Data : 
AP,AP50,AP75,APs,APm,APl
78.8223,85.2194,84.5590,-100.0000,0.0000,78.8323


MODEL IS LIVE: 

The model is live and running on a GPU server.
User can test the model by running test_app.py using python.
The user has to feed the path of the car image they want to test and the app will fetch the information from server.
Afer the script is run, a pop up window will appear that displays the results.

Instruction to run the model on ur local server:

1. User need to install the packages in serverCode/requirements.txt
2. Then user has to install detectron2 by following the instructions on https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md.
3. The the user has to run 'python serverCode/server.py' to run the server locally.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published