mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2024-12-15 02:11:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed slic for older versions of skimage

Browse Source
This commit is contained in:
Abubakar Abid 2021-07-12 15:13:18 -05:00
parent 11759cb3a7
commit f1d54145a8
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gradio/inputs.py
View File

@ -734,9 +734,14 @@ class Image(InputComponent):
from skimage.segmentation import slic
except (ImportError, ModuleNotFoundError):
raise ValueError("Error: running this interpretation for images requires scikit-image, please install it first.")
segments_slic = slic(
resized_and_cropped_image, self.interpretation_segments, compactness=10,
sigma=1, start_label=0)
try:
segments_slic = slic(
resized_and_cropped_image, self.interpretation_segments, compactness=10,
sigma=1, start_label=1)
except TypeError: # For skimage 0.16 and older
segments_slic = slic(
resized_and_cropped_image, self.interpretation_segments, compactness=10,
sigma=1)
return segments_slic, resized_and_cropped_image
def tokenize(self, x):