Print freeform output

Author: janjongboom

examples/audio/classify.py

@@ -52,11 +52,19 @@ def main(argv):
                 print("Device ID "+ str(selected_device_id) + " has been provided as an argument.")
 
             for res, audio in runner.classifier(device_id=selected_device_id):
-                print('Result (%d ms.) ' % (res['timing']['dsp'] + res['timing']['classification']), end='')
-                for label in labels:
-                    score = res['result']['classification'][label]
-                    print('%s: %.2f\t' % (label, score), end='')
-                print('', flush=True)
+                if "classification" in res["result"].keys():
+                    print('Result (%d ms.) ' % (res['timing']['dsp'] + res['timing']['classification']), end='')
+                    for label in labels:
+                        score = res['result']['classification'][label]
+                        print('%s: %.2f\t' % (label, score), end='')
+                    print('', flush=True)
+                elif "freeform" in res['result'].keys():
+                    print('Result (%d ms.)' % (res['timing']['dsp'] + res['timing']['classification']))
+                    for i in range(0, len(res['result']['freeform'])):
+                        print(f'    Freeform output {i}:', ", ".join(f"{x:.4f}" for x in res['result']['freeform'][i]))
+                else:
+                    print('Result (%d ms.)' % (res['timing']['dsp'] + res['timing']['classification']))
+                    print(res['result'])
 
         finally:
             if (runner):

examples/image/classify-full-frame.py

@@ -124,6 +124,10 @@ def print_classification(res, tag):
                         print('%s: Found %d bounding boxes (%d ms.)' % (tag, len(res["result"]["bounding_boxes"]), res['timing']['dsp'] + res['timing']['classification']))
                         for bb in res["result"]["bounding_boxes"]:
                             print('\t%s (%.2f): x=%d y=%d w=%d h=%d' % (bb['label'], bb['value'], bb['x'], bb['y'], bb['width'], bb['height']))
+                    elif "freeform" in res['result'].keys():
+                        print('Result (%d ms.)' % (res['timing']['dsp'] + res['timing']['classification']))
+                        for i in range(0, len(res['result']['freeform'])):
+                            print(f'    Freeform output {i}:', ", ".join(f"{x:.4f}" for x in res['result']['freeform'][i]))
 
                     if "visual_anomaly_grid" in res["result"].keys():
                         print('Found %d visual anomalies (%d ms.)' % (len(res["result"]["visual_anomaly_grid"]), res['timing']['dsp'] +

examples/image/classify-image.py

@@ -77,6 +77,11 @@ def main(argv):
                     print('\t%s (%.2f): x=%d y=%d w=%d h=%d' % (bb['label'], bb['value'], bb['x'], bb['y'], bb['width'], bb['height']))
                     cropped = cv2.rectangle(cropped, (bb['x'], bb['y']), (bb['x'] + bb['width'], bb['y'] + bb['height']), (255, 0, 0), 1)
 
+            elif "freeform" in res['result'].keys():
+                print('Result (%d ms.)' % (res['timing']['dsp'] + res['timing']['classification']))
+                for i in range(0, len(res['result']['freeform'])):
+                    print(f'    Freeform output {i}:', ", ".join(f"{x:.4f}" for x in res['result']['freeform'][i]))
+
             if "visual_anomaly_grid" in res["result"].keys():
                 print('Found %d visual anomalies (%d ms.)' % (len(res["result"]["visual_anomaly_grid"]), res['timing']['dsp'] +
                                                                                                             res['timing']['classification'] +

examples/image/classify-video.py

@@ -95,6 +95,11 @@ def getFrame(sec):
                         print('\t%s (%.2f): x=%d y=%d w=%d h=%d' % (bb['label'], bb['value'], bb['x'], bb['y'], bb['width'], bb['height']))
                         img = cv2.rectangle(cropped, (bb['x'], bb['y']), (bb['x'] + bb['width'], bb['y'] + bb['height']), (255, 0, 0), 1)
 
+                elif "freeform" in res['result'].keys():
+                    print('Result (%d ms.)' % (res['timing']['dsp'] + res['timing']['classification']))
+                    for i in range(0, len(res['result']['freeform'])):
+                        print(f'    Freeform output {i}:', ", ".join(f"{x:.4f}" for x in res['result']['freeform'][i]))
+
                 # Object tracking output
                 if "object_tracking" in res["result"].keys():
                     print('Found %d tracked objects' % (len(res["result"]["object_tracking"])))

examples/image/classify.py

@@ -118,6 +118,10 @@ def main(argv):
                     for bb in res["result"]["bounding_boxes"]:
                         print('\t%s (%.2f): x=%d y=%d w=%d h=%d' % (bb['label'], bb['value'], bb['x'], bb['y'], bb['width'], bb['height']))
                         img = cv2.rectangle(img, (bb['x'], bb['y']), (bb['x'] + bb['width'], bb['y'] + bb['height']), (255, 0, 0), 1)
+                elif "freeform" in res['result'].keys():
+                    print('Result (%d ms.)' % (res['timing']['dsp'] + res['timing']['classification']))
+                    for i in range(0, len(res['result']['freeform'])):
+                        print(f'    Freeform output {i}:', ", ".join(f"{x:.4f}" for x in res['result']['freeform'][i]))
 
                 if (show_camera):
                     cv2.imshow('edgeimpulse', cv2.cvtColor(img, cv2.COLOR_RGB2BGR))