Jordan-Pierce commited on
Commit
1482f3b
1 Parent(s): 794e3e3

Changed color mode

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -115,7 +115,7 @@ def run_inference(test_image):
115
  v_inf = Visualizer(img[:, :, ::-1],
116
  metadata=fathomnet_metadata,
117
  scale=1.0,
118
- instance_mode=ColorMode.IMAGE_BW)
119
 
120
  insts = []
121
 
 
115
  v_inf = Visualizer(img[:, :, ::-1],
116
  metadata=fathomnet_metadata,
117
  scale=1.0,
118
+ instance_mode=ColorMode.SEGMENTATION)
119
 
120
  insts = []
121