AdaBins:官方实施Adabins

阿达宾斯 正式实施 下载链接 您可以从下载经过预训练的模型“ AdaBins_nyu.pt”和“ AdaBins_kitti.pt” 您可以在下载NYU-Depth-v2官方测试集和KITTI Eigen拆分测试集的16位格式的预测深度 推理 将下载的砝码移至您选择的目录（此处将使用“ ./pretrained/”）。 然后，您可以像这样使用预训练的模型： from models import UnetAdaptiveBins import model_io from PIL import Image MIN_DEPTH = 1e-3 MAX_DEPTH_NYU = 10 MAX_DEPTH_KITTI = 80 N_BINS = 256 # NYU model = UnetAdaptiveBins . build ( n_bins = N_BINS , min_val = MIN