update

CodSpeedHQ · Sep 15, 2024 · aa707a5 · aa707a5
1 parent e486f29
commit aa707a5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/kornia/models/edge_detector/dexined.py b/kornia/models/edge_detector/dexined.py
@@ -26,10 +26,8 @@ def to_onnx(
  image_size: Optional[int] = 352,
  ) -> Tuple[str, EdgeDetector]:
  edge_detector = DexiNedBuilder.build(pretrained, image_size)
- model_name = "DexiNed"
  if onnx_name is None:
- _model_name = model_name
- onnx_name = f"Kornia-RTDETR-{_model_name}-{image_size}.onnx"
+ onnx_name = f"Kornia-DexiNed-{image_size}.onnx"
 
  if image_size is None:
  val_image = rand(1, 3, 352, 352)