↓ 1 callersFunctioncontext_upsample @disp_low: (b,1,h,w) 1/4 resolution @up_weights: (b,9,4*h,4*w) Image resolution
stereo/modeling/models/foundationstereo/core/submodule.py:490
↓ 1 callersFunctioncreate_module_dict(*, module, n_per_class_list, n_tries, nb_knn, train_features, train_labels)
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:198
↓ 1 callersFunctiondisp_warpWarping by disparity Args: img: [B, 3, H, W] disp: [B, 1, H, W], positive padding_mode: 'zeros' or 'border' Returns:
stereo/modeling/disp_refinement/disp_refinement.py:113
↓ 1 callersFunctiondisp_warpWarping by disparity Args: img: [B, 3, H, W] disp: [B, 1, H, W], positive padding_mode: 'zeros' or 'border' Returns:
stereo/modeling/models/aanet/submodule.py:801
↓ 1 callersFunctioneval_knn(
model,
train_dataset,
val_dataset,
accuracy_averaging,
nb_knn,
temperature,
batc
stereo/modeling/models/foundationstereo/dinov2/eval/knn.py:245
↓ 1 callersFunctioneval_linear(
*,
feature_model,
linear_classifiers,
train_data_loader,
val_data_loader,
metrics_fi
stereo/modeling/models/foundationstereo/dinov2/eval/linear.py:313
↓ 1 callersFunctionexport_engine(model, inputs, file, half, dynamic, simplify, optimize, workspace=4, verbose=False, prefix=colorstr('TensorRT
deploy/export.py:198