zayfod / pycozmo

A pure-Python communication library, alternative SDK, and application for the Cozmo robot.
MIT License
178 stars 59 forks source link

Face detection #49

Open davinellulinvega opened 3 years ago

davinellulinvega commented 3 years ago

So this is a work in progress, but the basics for object detection and tracking (for the moment only faces and hands are available) have been implemented in two separate classes: ObjectTracker and ObjectDetector. This will allow future scripts to take advantage of the Yolo V4 network hidden behind the ObjectDetector to detect more objects. The third class introduced here, namely ObjectDetectionNTracking is only here to hide the "complexities" of communicating with both the tracker and detector. Finally, I also implemented some kind of wrapper around the cv.imshow() function so that image display happens in another thread, and to make it hopefully easier for users to handle.