ahmetozlu / vehicle_counting_tensorflow

:oncoming_automobile: "MORE THAN VEHICLE COUNTING!" This project provides prediction for speed, color and size of the vehicles with TensorFlow Object Counting API.

Home Page:https://www.youtube.com/watch?v=PrqnhHf6fhM

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Ambiguity in visualize_boxes_and_labels_on_image_array() function

ChampionTej05 opened this issue · comments

As per the snippet from the documentation, visualize_boxes_and_labels_on_image_array function accepts arguments which are completely different from your own arguments. Also it returns the image as you can see below, then how are you supposed to collect the 'counter' and 'csv line' from that fucntion?

 Returns:

    uint8 numpy array with shape (img_height, img_width, 3) with overlaid boxes

I modified the code as per the arguments but now I am not able to find the count when the object crosses ROI. Please help.
MODIFIED SNIPPET


counter = len(boxes)
                print("Cureent running counter: ",counter)

                # Visualization of the results of a detection.

                input_frame = vis_util.visualize_boxes_and_labels_on_image_array(

                    input_frame,

                    np.squeeze(boxes),

                    np.squeeze(classes).astype(np.int32),

                    np.squeeze(scores),

                    category_index,

                    use_normalized_coordinates=True,

                    line_thickness=4)