tensorflow_object_counting_api
tensorflow_object_counting_api copied to clipboard
How to use vehicle_counting.py
Hi, the following files are my custom model to detect "mice" I also have the frozen.pb file although not sure if that's equivalent to frozen_inference_graph.pb
This is the .weights file I downloaded and converted through the repo before "EDIT 1": https://drive.google.com/file/d/1GKVQipCa9q3Vk10yF78AR2Ip0GOe_2Ib/view

what I want to do is count how many times the mouse crossed the line (a type of test analysis) but I'm not sure which parts of vehicle_counting.py to edit, specifically the 'mscoco_label_map.pbtxt' field in line 17, do I change it or keep it that way? specially if I have to add something like this in it: item { name: "/m/01g317" id: 1 display_name: "person" } I obviously don't know the "name" or is there a way to know that ?
Also if you are wondering how I got the custom model files I converted a custom yolov3.weights file from this repo: https://github.com/mystic123/tensorflow-yolo-v3 and it produced all the files in the screenshots
EDIT 1: I tried to run it using the frozen_inference_graph.pb file in the screenshots but this error happened:
Traceback (most recent call last):
File "d:/custom code/tensorflow_object_counting_api/vehicle_counting.py", line 18, in
ALSO I changed the .pbtxt file to pet_label_map.pbtxt and added an item in it: item { id: 38 name: 'mice' }