阿达宾斯
正式实施
下载链接
您可以从下载经过预训练的模型“ 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
2023-05-26 16:01:52
552KB
Python
1