remove cv2

.idea/workspace.xml

@@ -4,9 +4,9 @@
     <option name="autoReloadType" value="SELECTIVE" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="d7806539-b6d6-42e7-bb45-1565f5d54891" name="Changes" comment="init commit">
+    <list default="true" id="d7806539-b6d6-42e7-bb45-1565f5d54891" name="Changes" comment="fix build">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/Dockerfile" beforeDir="false" afterPath="$PROJECT_DIR$/Dockerfile" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/model.py" beforeDir="false" afterPath="$PROJECT_DIR$/model.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
@@ -119,7 +119,7 @@
       <option name="presentableId" value="Default" />
       <updated>1702301546100</updated>
       <workItem from="1702301554378" duration="4025000" />
-      <workItem from="1702450051021" duration="1074000" />
+      <workItem from="1702450051021" duration="2218000" />
     </task>
     <task id="LOCAL-00001" summary="init commit">
       <option name="closed" value="true" />
@@ -129,7 +129,15 @@
       <option name="project" value="LOCAL" />
       <updated>1702450645297</updated>
     </task>
-    <option name="localTasksCounter" value="2" />
+    <task id="LOCAL-00002" summary="fix build">
+      <option name="closed" value="true" />
+      <created>1702451179622</created>
+      <option name="number" value="00002" />
+      <option name="presentableId" value="LOCAL-00002" />
+      <option name="project" value="LOCAL" />
+      <updated>1702451179622</updated>
+    </task>
+    <option name="localTasksCounter" value="3" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
@@ -137,7 +145,8 @@
   </component>
   <component name="VcsManagerConfiguration">
     <MESSAGE value="init commit" />
-    <option name="LAST_COMMIT_MESSAGE" value="init commit" />
+    <MESSAGE value="fix build" />
+    <option name="LAST_COMMIT_MESSAGE" value="fix build" />
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
     <SUITE FILE_PATH="coverage/detector$demo.coverage" NAME="demo Coverage Results" MODIFIED="1702302380580" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />

model.py

@@ -1,9 +1,8 @@
-
-import cv2
 import torch
 import random
 from PIL import ImageDraw
 import torchvision.transforms as T
+from PIL.ImageFont import ImageFont, truetype
 
 # COCO Classes
 CLASSES = [
@@ -69,10 +68,9 @@ def plot_one_box(x, img, color=None, label=None, line_thickness=None):
     img_draw.rectangle((c1[0], c1[1], c2[0], c2[1]), outline=color, width=tl)
     if label:
         tf = max(tl - 1, 1)  # font thickness
-        t_size = cv2.getTextSize(label, 0, fontScale=tl / 5, thickness=tf)[0]
-        c2 = c1[0] + t_size[0], c1[1] + t_size[1] - 3
-        img_draw.rectangle((c1, c2), fill=color)
-        img_draw.text((c1[0], c1[1] - 2), label, fill=(255, 255, 255))
+        x1, y1, x2, y2 = img_draw.textbbox(c1, label, stroke_width=tf)
+        img_draw.rectangle((x1, y1, x2, y2), fill=color)
+        img_draw.text((x1, y1), label, fill=(255, 255, 255))
 
 
 # Ploting Bounding Box on img
@@ -91,8 +89,6 @@ def detect(im, confidence):
     model.eval()
 
     scores, boxes = image_processing(im, model, transform, confidence / 100)
-    # im = cv2.imread(image_path)
-    # im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
     im = add_bboxes(im, scores, boxes)
 
     return im

requirements.txt

@@ -26,7 +26,6 @@ mdurl==0.1.2
 mpmath==1.3.0
 networkx==3.2.1
 numpy==1.26.2
-opencv-python==4.8.1.78
 packaging==23.2
 pandas==2.1.4
 Pillow==10.1.0