- 博客(1)
- 收藏
- 关注
原创 AdaBins pt模型转换为onnx
def transform_pt_to_onnx(): MIN_DEPTH = 1e-3 MAX_DEPTH_NYU = 10 N_BINS = 256 batch_size = 1 input_shape = (3, 480, 640) pretrained_path = "./model/AdaBins_nyu.pt" onnx_output_path = "./model/AdaBins_nyu.onnx" device = t.
2021-10-18 20:37:38 661 4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人