- Put training images into folder .\roipoly_annotate\train
Put testing images into folder .\roipoly_annotate\test - Execute annotate.py to hand-label RedBarrel and TrickyCases. Remember to change colorClass in the code between. TrickyCases are basically light yellow ceiling, windows and dark red floor.
- Run train.py to get a GMM Model
- Once the model is obtained, run predict.py to detect red barrel. A image with bounding box and centroid will appear, also info like centroid, distance and numbers will be at command window.
-
Notifications
You must be signed in to change notification settings - Fork 0
MaxLing/gmm_color_object_detection
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
a hybrid Gaussian Mixture Model for color segmentation, and connected component analysis for object detection
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published