App bug fix.

app.py

@@ -13,6 +13,7 @@ from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
 
 from PIL import Image, ImageDraw
 import cv2
+import numpy
 import matplotlib.pyplot as plt
 
 id2label = {1: 'person', 2: 'rider', 3: 'car', 4: 'bus', 5: 'truck', 6: 'bike', 7: 'motor', 8: 'traffic light', 9: 'traffic sign', 10: 'train'}
@@ -100,20 +101,21 @@ def rescale_bboxes(out_bbox, size):
 
 def plot_results(pil_img, prob, boxes):
 
-    draw = ImageDraw.Draw(pil_img)
+    img = numpy.asarray(pil_img)
     
     for p, (xmin, ymin, xmax, ymax) in zip(prob, boxes.tolist()):
         cl = p.argmax()
         c = colors[cl]
+        c1, c2 = (xmin, ymin), (xmax, ymax)
 
-        draw.rectangle([xmin, ymin, xmax - xmin, ymax - ymin], outline=c, width=2)
-        draw.text(
-            [xmin + 5, ymin + 5], 
-            f'{id2label[cl.item()]}: {p[cl]:0.2f}',
-            fill=c)
+        cv2.rectangle(img, c1, c2, c, thickness=2, lineType=cv2.LINE_AA)
+        # cv2.text(
+        #     [xmin + 5, ymin + 5], 
+        #     f'{id2label[cl.item()]}: {p[cl]:0.2f}',
+        #     fill=c)
         # ax.text(xmin, ymin, text, fontsize=10,
         #         bbox=dict(facecolor=c, alpha=0.5))
-    return pil_img
+    return return Image.fromarray(img[:,:,::-1])
     # return fig
 
 

requirements.txt

@@ -6,4 +6,5 @@ matplotlib>=3.2.2
 Pillow>=7.1.2
 torch==1.10.0
 pytorch-lightning==1.9.3
-opencv-python>=4.1.1
+opencv-python>=4.1.1
+numpy>=1.18.5