cv2 headless :bomb:

app.py

@@ -1,5 +1,4 @@
 from PIL import Image
-import cv2
 import torch
 from numpy import random
 
@@ -63,7 +62,6 @@ def detect(pil_img,names):
             for *xyxy, conf, cls in det:
                 label = '%s %.2f' % (names[int(cls)], conf)
                 plot_one_box(xyxy, img_np, label=label, color=colors[int(cls)], line_thickness=3)
-    cv2.imwrite('/tmp/aaa.jpg',img_np[:,:,::-1])
     return Image.fromarray(img_np)
     
 

requirements.txt

@@ -4,7 +4,8 @@
 Cython
 matplotlib>=3.2.2
 numpy>=1.18.5
-opencv-python>=4.1.2
+# opencv-python>=4.1.2
+opencv-python-headless>=4.1.2
 Pillow
 PyYAML>=5.3.1
 scipy>=1.4.1