Spaces:
Runtime error
Runtime error
adding danger symbol
Browse files- .gitignore +2 -2
- inference.py +3 -3
- static/danger_symbol.jpeg +0 -0
- static/danger_symbol1.png +0 -0
.gitignore
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
# Added
|
2 |
tmp/
|
3 |
-
|
4 |
# Byte-compiled / optimized / DLL files
|
5 |
__pycache__/
|
6 |
*.py[cod]
|
@@ -166,4 +166,4 @@ cython_debug/
|
|
166 |
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
167 |
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
168 |
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
169 |
-
#.idea/
|
|
|
1 |
# Added
|
2 |
tmp/
|
3 |
+
|
4 |
# Byte-compiled / optimized / DLL files
|
5 |
__pycache__/
|
6 |
*.py[cod]
|
|
|
166 |
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
167 |
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
168 |
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
169 |
+
#.idea/
|
inference.py
CHANGED
@@ -52,7 +52,7 @@ classes = ['Beach',
|
|
52 |
'Whale',
|
53 |
'Dolphin',
|
54 |
'Miscellaneous',
|
55 |
-
'
|
56 |
'C Shark',
|
57 |
'Dusty shark',
|
58 |
'Blue shark',
|
@@ -105,7 +105,7 @@ class_sizes = {'Beach': None,
|
|
105 |
'Whale': {'feet':[15, 30], 'meter': [4.5, 10], 'kg': [2500, 80000], 'pounds': [55000, 176000]},
|
106 |
'Dolphin': {'feet':[6.6, 13.1], 'meter': [2, 4], 'kg': [150, 650], 'pounds': [330, 1430]},
|
107 |
'Miscellaneous': None,
|
108 |
-
'
|
109 |
'C Shark': {'feet': [4, 10], 'meter': [1.25, 3], 'kg': [50, 1000], 'pounds': [110, 2200]}, # Prob incorrect
|
110 |
'Dusty shark': {'feet': [9, 14], 'meter': [3, 4.25], 'kg': [160, 180], 'pounds': [350, 400]},
|
111 |
'Blue shark': {'feet': [7.9, 12.5], 'meter': [2.4, 3], 'kg': [60, 120], 'pounds': [130, 260]},
|
@@ -133,7 +133,7 @@ classes_is_human_id = [i for i, x in enumerate(classes_is_human) if x == 1]
|
|
133 |
classes_is_unknown_id = [i for i, x in enumerate(classes_is_unknown) if x == 1]
|
134 |
|
135 |
|
136 |
-
if not os.path.exists('model')
|
137 |
REPO_ID = "SharkSpace/maskformer_model"
|
138 |
FILENAME = "mask2former"
|
139 |
snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')
|
|
|
52 |
'Whale',
|
53 |
'Dolphin',
|
54 |
'Miscellaneous',
|
55 |
+
'Shadow',
|
56 |
'C Shark',
|
57 |
'Dusty shark',
|
58 |
'Blue shark',
|
|
|
105 |
'Whale': {'feet':[15, 30], 'meter': [4.5, 10], 'kg': [2500, 80000], 'pounds': [55000, 176000]},
|
106 |
'Dolphin': {'feet':[6.6, 13.1], 'meter': [2, 4], 'kg': [150, 650], 'pounds': [330, 1430]},
|
107 |
'Miscellaneous': None,
|
108 |
+
'Shadow': {'feet': [2, 15], 'meter': [0.6, 4.5], 'kg': [50, 1000], 'pounds': [110, 2200]},
|
109 |
'C Shark': {'feet': [4, 10], 'meter': [1.25, 3], 'kg': [50, 1000], 'pounds': [110, 2200]}, # Prob incorrect
|
110 |
'Dusty shark': {'feet': [9, 14], 'meter': [3, 4.25], 'kg': [160, 180], 'pounds': [350, 400]},
|
111 |
'Blue shark': {'feet': [7.9, 12.5], 'meter': [2.4, 3], 'kg': [60, 120], 'pounds': [130, 260]},
|
|
|
133 |
classes_is_unknown_id = [i for i, x in enumerate(classes_is_unknown) if x == 1]
|
134 |
|
135 |
|
136 |
+
if not os.path.exists('model'):
|
137 |
REPO_ID = "SharkSpace/maskformer_model"
|
138 |
FILENAME = "mask2former"
|
139 |
snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')
|
static/danger_symbol.jpeg
DELETED
Binary file (61.1 kB)
|
|
static/danger_symbol1.png
ADDED