blob: a0a691aa72a2b8c55249116fb5c571337d67784e [file] [log] [blame]
import vision
def run_detector_example():
detector = vision.Detector(vision.FACE_DETECTION_MODEL)
for frame in vision.get_frames('Face Detector', size=(640, 480)):
faces = detector.get_objects(frame)
vision.draw_objects(frame, faces)
def run_classifier_example():
labels = vision.load_labels(vision.CLASSIFICATION_LABELS)
classifier = vision.Classifier(vision.CLASSIFICATION_MODEL)
for frame in vision.get_frames('Object Classifier', size=(640, 480)):
classes = classifier.get_classes(frame)
vision.draw_classes(frame, classes, labels)
if __name__ == '__main__':
run_classifier_example()
#run_detector_example()