Spaces:
Sleeping
Sleeping
Code update for inference on IT107 Assignment
Browse files- app.py +9 -6
- data/label_map.pbtxt +5 -1
- test_samples/sample_balloon.jpeg +0 -0
app.py
CHANGED
@@ -7,10 +7,11 @@ from object_detection.utils import label_map_util
|
|
7 |
from object_detection.utils import visualization_utils as viz_utils
|
8 |
from object_detection.utils import ops as utils_op
|
9 |
import tarfile
|
10 |
-
import
|
|
|
11 |
import gradio as gr
|
12 |
|
13 |
-
|
14 |
PATH_TO_LABELS = 'data/label_map.pbtxt'
|
15 |
category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
|
16 |
|
@@ -28,10 +29,12 @@ def load_image_into_numpy_array(path):
|
|
28 |
return pil_image_as_numpy_array(image)
|
29 |
|
30 |
def load_model():
|
31 |
-
wget.download("https://nyp-aicourse.s3-ap-southeast-1.amazonaws.com/pretrained-models/balloon_model.tar.gz")
|
32 |
-
tarfile.open("balloon_model.tar.gz").extractall()
|
33 |
-
|
34 |
-
|
|
|
|
|
35 |
return detection_model
|
36 |
|
37 |
# samples_folder = 'test_samples
|
|
|
7 |
from object_detection.utils import visualization_utils as viz_utils
|
8 |
from object_detection.utils import ops as utils_op
|
9 |
import tarfile
|
10 |
+
from huggingface_hub import snapshot_download
|
11 |
+
import os
|
12 |
import gradio as gr
|
13 |
|
14 |
+
MODEL_REPO = 'sokonana/it107model'
|
15 |
PATH_TO_LABELS = 'data/label_map.pbtxt'
|
16 |
category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
|
17 |
|
|
|
29 |
return pil_image_as_numpy_array(image)
|
30 |
|
31 |
def load_model():
|
32 |
+
# wget.download("https://nyp-aicourse.s3-ap-southeast-1.amazonaws.com/pretrained-models/balloon_model.tar.gz")
|
33 |
+
# tarfile.open("balloon_model.tar.gz").extractall()
|
34 |
+
model_path = snapshot_download(MODEL_REPO)
|
35 |
+
|
36 |
+
model_dir = os.path.join(model_path, 'saved_model')
|
37 |
+
detection_model = tf.saved_model.load(model_dir)
|
38 |
return detection_model
|
39 |
|
40 |
# samples_folder = 'test_samples
|
data/label_map.pbtxt
CHANGED
@@ -1,4 +1,8 @@
|
|
1 |
item {
|
2 |
id: 1
|
3 |
-
name: '
|
|
|
|
|
|
|
|
|
4 |
}
|
|
|
1 |
item {
|
2 |
id: 1
|
3 |
+
name: 'Giant panda'
|
4 |
+
}
|
5 |
+
item {
|
6 |
+
id: 2
|
7 |
+
name: 'Red panda'
|
8 |
}
|
test_samples/sample_balloon.jpeg
DELETED
Binary file (38.9 kB)
|
|