mask -> mask_person

agermanidis · Jul 18, 2019 · bc67e80 · bc67e80
1 parent 6abf221
commit bc67e80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/runway_model.py b/runway_model.py
@@ -41,7 +41,7 @@ def segment(model, inputs):
   return seg_map.astype(np.uint8)
 
 
-@runway.command('mask', inputs={'image': runway.image}, outputs={'masked_image': runway.image})
+@runway.command('mask_person', inputs={'image': runway.image}, outputs={'masked_image': runway.image})
 def mask(model, inputs):
   _, seg_map = model.run(inputs['image'])
   mask = np.stack((seg_map,)*4, axis=-1)
@@ -52,4 +52,4 @@ def mask(model, inputs):
 
 
 if __name__ == '__main__':
-  runway.run(port=5232)
+  runway.run(port=5232)