Running Retinanet Object detector on tensorflow using a pretrained pb file. (Ask me for the pb file)
- Tensorflow (tested with r1.4.0)
- Numpy
- Opencv (tested with 3.3)
- pickle
- Download the pretrained model's pb file and place it in model_files.
- Open pb_inference.py and change the name of the variable 'IMAGE_PATH' to point to the absolute path of your test image.
- Run the script.
The code is setup to output a csv file containing all the labels of the objects in the image and the corresponding x1, y1, x2 and y2 position. The xi and yi positions are in the form of ratios of width and height of the image. Point (0,0) corresponds to the top-left corner of the image.