Merge pull request #99 from seth-planet/seg_name_lists

Use new TPU segment option

Author: ChrisMaunder

src/modules/ObjectDetectionCoral/objectdetection_coral_multitpu.py

@@ -244,7 +244,7 @@ def main():
           tot_infr_time += infr_time
 
           # Start a timer for the last ~half of the run for more accurate benchmark
-          if chunk_i > (args.count-1) / 3.0:
+          if chunk_i > (args.count-1) / 2.0:
             half_infr_count += 1
             if half_wall_start is None:
               half_wall_start = time.perf_counter()
@@ -271,25 +271,25 @@ def main():
   if half_wall_start is not None:
     half_wall_time = time.perf_counter() - half_wall_start
 
-  print('completed one run every %.2fms for %d runs; %.2fms wall time for a single run' %
+  logging.info('completed one run every %.2fms for %d runs; %.2fms wall time for a single run' %
                             (wall_time * 1000 / args.count, args.count,
                             (time.perf_counter() - start_one) * 1000))
 
-  print('%.2fms avg time blocked across %d threads; %.2fms ea for final %d inferences' %
+  logging.info('%.2fms avg time blocked across %d threads; %.3fms ea for final %d inferences' %
                             (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')