Spaces:
Sleeping
Sleeping
Commit
·
b67ec45
1
Parent(s):
c605bde
vocab added
Browse files
app.ipynb
CHANGED
|
@@ -140,29 +140,7 @@
|
|
| 140 |
"outputs": [],
|
| 141 |
"source": [
|
| 142 |
"#!export\n",
|
| 143 |
-
"equipment_labels =
|
| 144 |
-
" 'Archery Bow',\n",
|
| 145 |
-
" 'Badminton Shuttlecock',\n",
|
| 146 |
-
" 'Baseball Bat',\n",
|
| 147 |
-
" 'Basketball ball',\n",
|
| 148 |
-
" 'Bowling Ball',\n",
|
| 149 |
-
" 'Boxing Gloves',\n",
|
| 150 |
-
" 'Carrom board',\n",
|
| 151 |
-
" 'Chessboard',\n",
|
| 152 |
-
" 'Cricket Bat',\n",
|
| 153 |
-
" 'Frisbee disc',\n",
|
| 154 |
-
" 'Golf ball',\n",
|
| 155 |
-
" 'Hockey Stick',\n",
|
| 156 |
-
" 'Ice Skates',\n",
|
| 157 |
-
" 'Rugby Ball',\n",
|
| 158 |
-
" 'Skateboard',\n",
|
| 159 |
-
" 'Ski Poles',\n",
|
| 160 |
-
" 'Soccer ball',\n",
|
| 161 |
-
" 'Table Tennis Paddle',\n",
|
| 162 |
-
" 'Tennis Racket',\n",
|
| 163 |
-
" 'Volleyball ball'\n",
|
| 164 |
-
"\n",
|
| 165 |
-
")\n",
|
| 166 |
"\n",
|
| 167 |
"def recognize_image(image):\n",
|
| 168 |
" pred, idx, probs = model.predict(image)\n",
|
|
|
|
| 140 |
"outputs": [],
|
| 141 |
"source": [
|
| 142 |
"#!export\n",
|
| 143 |
+
"equipment_labels = model.dls.vocab\n",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 144 |
"\n",
|
| 145 |
"def recognize_image(image):\n",
|
| 146 |
" pred, idx, probs = model.predict(image)\n",
|