models.depth_anything_v2.depth_anything_v2_onnx
Source: models/depth_anything_v2/depth_anything_v2_onnx.py
models.depth_anything_v2.depth_anything_v2_onnx
ONNXCompatibleGlobalPool2d
Bases: Module
ONNX-compatible replacement for AdaptiveMaxPool2d(1) and AdaptiveAvgPool2d(1)
replace_adaptive_pooling_with_onnx_compatible(module)
Recursively replace AdaptiveMaxPool2d and AdaptiveAvgPool2d with ONNX-compatible versions