EasyOCR-Trainer

app.py

@@ -39,15 +39,15 @@ def calculate_iou(bbox1, bbox2):
     return iou
 
 
-cropped_dir = "./cropped_images/"
+cropped_dir = "./mnt/cropped_images/"
 if os.path.exists(cropped_dir):
         shutil.rmtree(cropped_dir)
 os.makedirs(cropped_dir, exist_ok=True)
 
-output_dir1 = "./Folder1"
-output_dir2 = "./Folder2"
-output_dir3 = "./Folder3"
-UPLOAD_FOLDER = "./data1"
+output_dir1 = "./mnt/Folder1"
+output_dir2 = "./mnt/Folder2"
+output_dir3 = "./mnt/Folder3"
+UPLOAD_FOLDER = "./mnt/data1"
 os.makedirs(output_dir1, exist_ok=True)
 os.makedirs(output_dir2, exist_ok=True)
 os.makedirs(output_dir3, exist_ok=True)
@@ -101,7 +101,7 @@ def process_image():
         detections = sv.Detections.from_ultralytics(results)
         annotated_image = bounding_box_annotator.annotate(scene=random_image_data, detections=detections)
         annotated_image = label_annotator.annotate(scene=annotated_image, detections=detections)
-        save_path = "./Folder1/" + "detection" + ii
+        save_path = "./mnt/Folder1/" + "detection" + ii
         cv2.imwrite(save_path, annotated_image)
         print(f"Annotated image saved at {save_path}")
         bounding_boxes = results.boxes.xyxy.cpu().numpy()