search

nicholaskajoh / ivy

Video-based object counting software.
MIT License
428 stars 170 forks source link

Adding a custom detector #51

Closed spranjal25 closed 4 years ago

spranjal25 commented 4 years ago

How can we add a custom detector to this module? I have a faster RCNN based detector which I added as a python file having the exact get_bounding_boxes() function which returns the bounding boxes, scores and the classes. I added it in the detectors directory but when I run the main.py file it gives: "ERROR : Invalid detector model, algorithm or API specified (options: yolo, tfoda, detectron2, haarcascade) {'label': 'INVALID_DETECTION_ALGORITHM'}" I checked in the .env file but I cannot seem to find what path may be added for the detector to run my fasterrcnn.py file instead of the given yolo.py, tfoda.py etc. Could you help me out with this one?

nicholaskajoh commented 4 years ago

Here: https://github.com/nicholaskajoh/ivy/blob/master/detectors/detector.py

spranjal25 commented 4 years ago

Thanks a ton! still wondering how I missed that :o.