added weights and testing updated to avoid installation error

hotosm · Oct 22, 2024 · f33b94d · f33b94d
1 parent 67c28c8
commit f33b94d
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -17,7 +17,7 @@ jobs:
           python-version: 3.8
 
       - name: Remove solaris
-        run: sudo rm -rf ./docker
+        run: sudo rm -rf ./docker ./weights
 
       - name: Clone Ramp
         run: git clone https://github.com/kshitijrajsharma/ramp-code-fAIr.git ramp-code

diff --git a/test_yolo.py b/test_yolo.py
@@ -59,7 +59,7 @@ def __exit__(self, type, value, traceback):
 
 output_path = train_yolo(
     data=f"{base_path}",
-    weights=f"{os.getcwd()}/weights/yolov8alb/best.pt",
+    weights=f"{os.getcwd()}/ylov8_seg_alb_best.pt",
     gpu="cpu",
     epochs=2,
     batch_size=16,

diff --git a/yolov8alb_weights/best.pt → ylov8_seg_alb_best.pt b/yolov8alb_weights/best.pt → ylov8_seg_alb_best.pt