sampreets3 / treecounter-ml Goto Github PK

When the program exits, we have an exception message saying that the canberra-gtk-module is not installed. This does not affect performance, but this Exception needs to be handled.

Have to retrain the entire network to be able to recognise posts, and count the distance between trees as well.
Be careful to take into account both the wooden posts and the metallic posts.

As of now, the program is simply displaying the frame count of the video. Need to add the tree counting feature to the program.
Basic algorithm would be :-

• Detect the centroid for each box
• Use the centroid against a demarcating line
• If the centroid is above the demarcating line, do nothing
• If the centroid is below the demarcating line, increment the tree counting variable

Need to add command-line argument passing feature. Probable list of options could be:

• --input: To specify the location of the input video file
• --output: To specify the location of the output video file
• --config: To specify the location of the model's config file
• --weights: To specify the location of the model's weights file
• --classes: To specify the location of the model's class descriptors

Can be fixed by using the argparse module in Python.