Upload 5 files

data.yaml

@@ -0,0 +1,6 @@
+train: ./dataset/autosplit_train.txt
+val: ./dataset/autosplit_val.txt
+test: ./dataset/autosplit_test.txt
+nc: 1
+names:
+- cell

requirements.txt

@@ -0,0 +1,2 @@
+ultralytics
+gradio

split.py

@@ -0,0 +1,7 @@
+from ultralytics.data.utils import autosplit
+
+autosplit(  
+    path="./dataset/images",
+    weights=(0.8, 0.1, 0.1),  # (train, validation, test) fractional splits
+    annotated_only=False,  # split only images with annotation file when True
+)

train.py

@@ -0,0 +1,6 @@
+from ultralytics import YOLO
+
+
+model = YOLO("yolov8n.pt")
+model.to('cuda')
+model.train(data="data.yaml", epochs=100, batch=8, device="cuda", imgsz=640)

val.py

@@ -0,0 +1,6 @@
+from ultralytics import YOLO
+
+
+model = YOLO("best_BCCM.pt")
+model.to('cuda')
+model.val(data="data.yaml")