Skip to content

Commit fd3f15b

Browse files
committed
fix: replace all hardcoded inferno fallbacks with viridis in config.get() calls
1 parent 916a5a7 commit fd3f15b

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

skills/transformation/depth-estimation/scripts/transform.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def parse_extra_args(self, parser: argparse.ArgumentParser):
149149

150150
def load_model(self, config: dict) -> dict:
151151
model_name = config.get("model", "depth-anything-v2-small")
152-
self.colormap_id = COLORMAP_MAP.get(config.get("colormap", "inferno"), 1)
152+
self.colormap_id = COLORMAP_MAP.get(config.get("colormap", "viridis"), 16)
153153
self.opacity = config.get("opacity", 0.5)
154154
self.blend_mode = config.get("blend_mode", "depth_only")
155155

@@ -204,7 +204,7 @@ def _load_coreml(self, config: dict) -> dict:
204204
"model": f"coreml-{variant_id}",
205205
"device": "neural_engine",
206206
"blend_mode": self.blend_mode,
207-
"colormap": config.get("colormap", "inferno"),
207+
"colormap": config.get("colormap", "viridis"),
208208
"backend": "coreml",
209209
}
210210

@@ -319,7 +319,7 @@ def _load_onnx(self, model_name: str, config: dict) -> dict:
319319
"model": model_name,
320320
"device": device_name,
321321
"blend_mode": self.blend_mode,
322-
"colormap": config.get("colormap", "inferno"),
322+
"colormap": config.get("colormap", "viridis"),
323323
"backend": "onnx",
324324
"execution_provider": active_ep,
325325
}
@@ -372,7 +372,7 @@ def _load_tensorrt(self, model_name: str, config: dict) -> dict:
372372
"model": model_name,
373373
"device": "cuda",
374374
"blend_mode": self.blend_mode,
375-
"colormap": config.get("colormap", "inferno"),
375+
"colormap": config.get("colormap", "viridis"),
376376
"backend": "tensorrt",
377377
}
378378

@@ -491,7 +491,7 @@ def _load_pytorch(self, model_name: str, config: dict) -> dict:
491491
"model": model_name,
492492
"device": self.device,
493493
"blend_mode": self.blend_mode,
494-
"colormap": config.get("colormap", "inferno"),
494+
"colormap": config.get("colormap", "viridis"),
495495
"backend": "pytorch",
496496
}
497497

0 commit comments

Comments
 (0)