piperod91 commited on
Commit
5818bcb
·
1 Parent(s): 81bc384

adding danger symbol

Browse files
.gitignore CHANGED
@@ -1,6 +1,6 @@
1
  # Added
2
  tmp/
3
- *.mp4
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
- 'Unidentifiable shark',
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
- 'Unidentifiable shark': {'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,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') and not os.getenv('SYSTEM') == 'spaces':
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