Kasamuday commited on
Commit
9183759
1 Parent(s): eb515e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ 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
  }
18
 
19
  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
+ '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'))