Spaces:
Sleeping
Sleeping
Henry Scheible
commited on
Commit
·
655ba4a
1
Parent(s):
6559107
add bounding box detector failure edge case
Browse files
app.py
CHANGED
@@ -99,6 +99,9 @@ def count_barnacles(image_raw, progress=gr.Progress()):
|
|
99 |
|
100 |
corners = wireframe_extractor(image_raw)
|
101 |
print(corners) # (0, 0, 1254, 1152)
|
|
|
|
|
|
|
102 |
|
103 |
cropped_image = image_raw[corners[1]:corners[3]+corners[1], corners[0]:corners[2]+corners[0], :]
|
104 |
|
|
|
99 |
|
100 |
corners = wireframe_extractor(image_raw)
|
101 |
print(corners) # (0, 0, 1254, 1152)
|
102 |
+
|
103 |
+
if corners[2] < 4 or corners[3] < 4:
|
104 |
+
return None, 0, []
|
105 |
|
106 |
cropped_image = image_raw[corners[1]:corners[3]+corners[1], corners[0]:corners[2]+corners[0], :]
|
107 |
|
upload.py
CHANGED
@@ -1,11 +1,13 @@
|
|
1 |
import wandb
|
2 |
|
|
|
|
|
3 |
run = wandb.init(
|
4 |
project="barnacles",
|
5 |
job_type='upload-model'
|
6 |
)
|
7 |
|
8 |
-
artifact = wandb.Artifact(name="
|
9 |
|
10 |
artifact.add_file("./sam_vit_h_4b8939.pth")
|
11 |
artifact.add_file("./requirements.txt")
|
|
|
1 |
import wandb
|
2 |
|
3 |
+
wandb.login()
|
4 |
+
|
5 |
run = wandb.init(
|
6 |
project="barnacles",
|
7 |
job_type='upload-model'
|
8 |
)
|
9 |
|
10 |
+
artifact = wandb.Artifact(name="sam_base", type="model")
|
11 |
|
12 |
artifact.add_file("./sam_vit_h_4b8939.pth")
|
13 |
artifact.add_file("./requirements.txt")
|