Update app.py
Browse files
app.py
CHANGED
@@ -13,8 +13,7 @@ from object_detection.utils import config_util
|
|
13 |
CUSTOM_MODEL_NAME = 'my_ssd_mobnet'
|
14 |
paths = {
|
15 |
'CHECKPOINT_PATH': os.path.join('Tensorflow', 'workspace', 'models', CUSTOM_MODEL_NAME),
|
16 |
-
'LABELMAP': os.path.join('Tensorflow', 'workspace', 'annotations', 'label_map.pbtxt')
|
17 |
-
'object_detection': os.path.join('Tensorflow', 'models', 'research', 'object_detection')
|
18 |
}
|
19 |
|
20 |
configs = config_util.get_configs_from_pipeline_file(os.path.join(paths['CHECKPOINT_PATH'], 'pipeline.config'))
|
|
|
13 |
CUSTOM_MODEL_NAME = 'my_ssd_mobnet'
|
14 |
paths = {
|
15 |
'CHECKPOINT_PATH': os.path.join('Tensorflow', 'workspace', 'models', CUSTOM_MODEL_NAME),
|
16 |
+
'LABELMAP': os.path.join('Tensorflow', 'workspace', 'annotations', 'label_map.pbtxt')
|
|
|
17 |
}
|
18 |
|
19 |
configs = config_util.get_configs_from_pipeline_file(os.path.join(paths['CHECKPOINT_PATH'], 'pipeline.config'))
|