minor tweaks

Author: mailseth

src/modules/ObjectDetectionCoral/objectdetection_coral_multitpu.py

@@ -279,17 +279,17 @@ def main():
                             (tot_infr_time / args.count, thread_cnt,
                              half_wall_time * 1000 / half_infr_count, half_infr_count))
 
-  print('-------RESULTS--------')
+  logging.info('-------RESULTS--------')
   if not objs:
-    print('No objects detected')
+    logging.info('No objects detected')
     return
 
   if any(objs):
     for obj in objs:
-      print(_tpu_runner.labels.get(obj.id, obj.id))
-      print('  id:    ', obj.id)
-      print('  score: ', obj.score)
-      print('  bbox:  ', obj.bbox)
+      logging.info(_tpu_runner.labels.get(obj.id, obj.id))
+      logging.info(f'  id:    {obj.id}')
+      logging.info(f'  score: {obj.score}')
+      logging.info(f'  bbox:  {obj.bbox}')
 
   if args.output:
     image = image.convert('RGB')

src/modules/ObjectDetectionCoral/tpu_runner.py

@@ -517,10 +517,12 @@ def _get_model_filenames(self, options: Options, tpu_list: list) -> list:
         if not any(tpu_list):
             return []
 
+        device_count  = len(tpu_list)  # TPUs. We've at least found one
         self.device_type   = 'Multi-TPU'
+        if device_count == 1:
+            self.device_type   = 'TPU'
 
         # If TPU found then default is single TPU model file (no segments)
-        device_count  = len(tpu_list)  # TPUs. We've at least found one
         if not any(options.tpu_segments_lists) or device_count == 1:
             return [options.model_tpu_file]