Move colors import

Author: giswqs

pyproject.toml

@@ -78,7 +78,7 @@ text = [
     "groundingdino-py",
 ]
 all = [
-    "segment-geospatial[fast, hq, fer, samgeo, samgeo2, text]",
+    "segment-geospatial[fast, hq, fer, samgeo, samgeo2, samgeo3, text]",
 ]
 extra = [
     "leafmap",

samgeo/samgeo3.py

@@ -476,6 +476,8 @@ def show_anns(
             # Fallback to image dimensions
             w, h = img_pil.size
 
+        COLORS = generate_colors(n_colors=128, n_samples=5000)
+
         for i in range(nb_objects):
             color = COLORS[i % len(COLORS)]
             plot_mask(results["masks"][i].squeeze(0).cpu(), color=color)
@@ -638,9 +640,6 @@ def generate_colors(n_colors=256, n_samples=5000):
     return colors_rgb
 
 
-COLORS = generate_colors(n_colors=128, n_samples=5000)
-
-
 def plot_bbox(
     img_height,
     img_width,