asmaamirkhan / TFObjectDetection

๐Ÿ‘ฉโ€๐Ÿ”ฌ Core code of object detection in Tensorflow (can be used to test models)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

๐Ÿ‘ฉโ€๐Ÿ”ฌ TensorFlow Object Detection

Core code of object detection in Tensorflow, can be used to test models.

๐Ÿ™Œ Available Codes

  1. detect_in_image: Detects or objects in a given image due to given Tensorflow model
  2. detect_in_video: Detects and objects in a given video due to given Tensorflow model

๐Ÿ‘ฎโ€โ™€๏ธ Make sure that you have OpenCV and Tensorflow already installed

๐Ÿ‘ฉโ€๐Ÿš€ Usage

  1. Clone or download this repo
  2. Open src folder in CMD

For detect_in_image.py:

  1. Run:
   python detect_in_image.py --input_image <PATH_TO_INPUT_JPG_FILE> --output_image <PATH_TO_OUTPUT_JPG_FILE>  --model_path <PATH_TO_INPUT_PB_FILE> --threshold <THRESHOLD>

For detect_in_video.py:

  1. Run:
python detect_in_video.py --input_video <PATH_TO_INPUT_MP4_FILE> --output_image <PATH_TO_OUTPUT_MP4_FILE>  --model_path <PATH_TO_INPUT_PB_FILE> --threshold <THRESHOLD>
  1. TADAA ๐ŸŽ‰ It's done ๐Ÿค—
  2. Press Q to stop

To see running options run for all codes: python detect_in_video.py --help

๐Ÿ‘ฎโ€โ™€๏ธ Thrshold value should be between 0 and 1.

๐Ÿ—๏ธ Used by

๐Ÿ’ผ Contact

Find me on LinkedIn and feel free to mail me, Asmaa ๐Ÿฆ‹

About

๐Ÿ‘ฉโ€๐Ÿ”ฌ Core code of object detection in Tensorflow (can be used to test models)

License:MIT License


Languages

Language:Python 100.0%