Kasamuday commited on
Commit
68b6675
1 Parent(s): 9183759

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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'))