Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ import cv2
|
|
9 |
import numpy as np
|
10 |
from PIL import Image
|
11 |
import gradio as gr
|
12 |
-
from roboflow import Roboflow
|
13 |
|
14 |
|
15 |
def parse_roboflow_url(url: str):
|
@@ -159,7 +159,7 @@ def upload_and_train_detection(
|
|
159 |
rf = Roboflow(api_key=api_key)
|
160 |
ws = rf.workspace(workspace)
|
161 |
|
162 |
-
#
|
163 |
try:
|
164 |
proj = ws.project(project_slug)
|
165 |
except Exception as e:
|
@@ -173,7 +173,18 @@ def upload_and_train_detection(
|
|
173 |
else:
|
174 |
raise
|
175 |
|
176 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
177 |
ws.upload_dataset(
|
178 |
dataset_path,
|
179 |
project_slug,
|
@@ -181,6 +192,7 @@ def upload_and_train_detection(
|
|
181 |
project_type=project_type
|
182 |
)
|
183 |
|
|
|
184 |
version_num = proj.generate_version(settings={
|
185 |
"augmentation": {},
|
186 |
"preprocessing": {},
|
|
|
9 |
import numpy as np
|
10 |
from PIL import Image
|
11 |
import gradio as gr
|
12 |
+
from roboflow import Roboflow
|
13 |
|
14 |
|
15 |
def parse_roboflow_url(url: str):
|
|
|
159 |
rf = Roboflow(api_key=api_key)
|
160 |
ws = rf.workspace(workspace)
|
161 |
|
162 |
+
# 1) Try to fetch existing project
|
163 |
try:
|
164 |
proj = ws.project(project_slug)
|
165 |
except Exception as e:
|
|
|
173 |
else:
|
174 |
raise
|
175 |
|
176 |
+
# 2) If it exists but is NOT object-detection, make a fresh slug
|
177 |
+
if getattr(proj, "annotation", None) != project_type:
|
178 |
+
new_slug = project_slug + "-v2"
|
179 |
+
proj = ws.create_project(
|
180 |
+
new_slug,
|
181 |
+
annotation=project_type,
|
182 |
+
project_type=project_type,
|
183 |
+
project_license=project_license
|
184 |
+
)
|
185 |
+
project_slug = new_slug
|
186 |
+
|
187 |
+
# 3) Upload train/val/test
|
188 |
ws.upload_dataset(
|
189 |
dataset_path,
|
190 |
project_slug,
|
|
|
192 |
project_type=project_type
|
193 |
)
|
194 |
|
195 |
+
# 4) Generate new version & train
|
196 |
version_num = proj.generate_version(settings={
|
197 |
"augmentation": {},
|
198 |
"preprocessing": {},
|