Fix bugs for apple silicon mac

i referred this issue hunglc007/tensorflow-yolov4-tflite#368

Author: Qone2

conda-gpu-apple-silicon-mac.yml

@@ -2,6 +2,7 @@ name: yolov4-gpu
 
 channels:
   - apple
+  - conda-forge
 dependencies:
   - tensorflow-deps
   - matplotlib
@@ -13,6 +14,7 @@ dependencies:
   - pillow
   - waitress
   - paste
+  - easydic
   - pip:
       - tensorflow-macos
       - tensorflow-metal

core/utils.py

@@ -155,16 +155,16 @@ def draw_bbox(image, bboxes, classes=read_class_names(cfg.YOLO.CLASSES), allowed
         else:
             bbox_color = colors[class_ind]
             bbox_thick = int(0.6 * (image_h + image_w) / 600)
-            c1, c2 = (coor[1], coor[0]), (coor[3], coor[2])
+            c1, c2 = (int(coor[1]), int(coor[0])), (int(coor[3]), int(coor[2]))
             cv2.rectangle(image, c1, c2, bbox_color, bbox_thick)
 
             if show_label:
                 bbox_mess = '%s: %.2f' % (classes[class_ind], score)
                 t_size = cv2.getTextSize(bbox_mess, 0, fontScale, thickness=bbox_thick // 2)[0]
                 c3 = (c1[0] + t_size[0], c1[1] - t_size[1] - 3)
-                cv2.rectangle(image, c1, (np.float32(c3[0]), np.float32(c3[1])), bbox_color, -1) #filled
+                cv2.rectangle(image, c1, (int(np.float32(c3[0])), int(np.float32(c3[1]))), bbox_color, -1) #filled
 
-                cv2.putText(image, bbox_mess, (c1[0], np.float32(c1[1] - 2)), cv2.FONT_HERSHEY_SIMPLEX,
+                cv2.putText(image, bbox_mess, (c1[0], int(np.float32(c1[1] - 2))), cv2.FONT_HERSHEY_SIMPLEX,
                             fontScale, (0, 0, 0), bbox_thick // 2, lineType=cv2.LINE_AA)
     return image