-
0 votesanswersviews
用于pytorch模型的Coreml模型浮点输入
我有一个pytorch模型,它将 3 x width x height 图像作为输入,像素值在 0-1 之间归一化 例如,在pytorch中输入 img = io.imread(img_path) input_img = torch.from_numpy( np.transpose(img, (2,0,1)) ).contiguous().float()/255.0 我将这个模型转换为core...