fix: update

fengx_launch.ipynb

@@ -33,21 +33,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Enumerating objects: 9, done.\n",
-      "Counting objects: 100% (9/9), done.\n",
+      "Enumerating objects: 7, done.\n",
+      "Counting objects: 100% (7/7), done.\n",
       "Delta compression using up to 8 threads\n",
-      "Compressing objects: 100% (5/5), done.\n",
-      "Writing objects: 100% (5/5), 426 bytes | 426.00 KiB/s, done.\n",
-      "Total 5 (delta 4), reused 0 (delta 0), pack-reused 0\n",
+      "Compressing objects: 100% (4/4), done.\n",
+      "Writing objects: 100% (4/4), 345 bytes | 345.00 KiB/s, done.\n",
+      "Total 4 (delta 3), reused 0 (delta 0), pack-reused 0\n",
       "To https://huggingface.co/spaces/rjx/rjxai_image_identification\n",
-      "   adea420..4eae44a  main -> main\n"
+      "   276ebff..4dcf678  main -> main\n"
      ]
     }
    ],

fengxai/SelfSupervised.py

@@ -2,6 +2,7 @@ import torch
 import supervision as sv
 import cv2
 import numpy as np
+import os
 from segment_anything import SamPredictor, sam_model_registry
 from diffusers import StableDiffusionInpaintPipeline
 from torchvision.ops import box_convert
@@ -66,10 +67,16 @@ class SelfSupervised:
         
         boxesHeight=int(xyxy[0][3]-xyxy[0][1])
         boxesWidth=int(xyxy[0][2]-xyxy[0][0])
-        cv2.imwrite("data/annotated_image.jpg", imgAnnnotated)
+        
+        imageOutPutFile = "data/annotated_image.jpg"
+        fileList = imageOutPutFile.split("/")[0]
+        if not fileList:
+            os.mkdir(fileList)
+        cv2.imwrite(imageOutPutFile, imgAnnnotated)
+        print("os cwd=", os.getcwd())
         
         return {
-            "imageOutput": "data/annotated_image.jpg",
+            "imageOutput": imageOutPutFile,
             "imageHeight": h,
             "imageWidth": w,
             "objectHeight": boxesHeight,