Henry Scheible commited on
Commit
655ba4a
·
1 Parent(s): 6559107

add bounding box detector failure edge case

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. upload.py +3 -1
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="barnacles/sam_base", type="model")
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")