Spaces:

TheOneReborn
/

car-segmentation

Sleeping

TheOneReborn commited on Mar 6

Commit

8103efe

1 Parent(s): 6d3ae22

fix: typo in method

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ def segment_image(image):
     prediction_resized = numpy.array(prediction_resized)
     # Apply a colormap for visualization (using the public API)
-    colormap = cm.colormaps['jet']
     # Normalize the mask and apply the colormap (result is in float [0,1])
     colored_mask = colormap(prediction_resized / numpy.max(prediction_resized))[:, :, :3]

     prediction_resized = numpy.array(prediction_resized)
     # Apply a colormap for visualization (using the public API)
+    colormap = cm._colormaps['jet']
     # Normalize the mask and apply the colormap (result is in float [0,1])
     colored_mask = colormap(prediction_resized / numpy.max(prediction_resized))[:, :, :3]