@@ -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