zsflalala

学长你好，数据预处理后，image图片大小是3，304，328，depth图片大小是4，114，152，我想问的第一个问题是，这些通道是从哪个函数开始改变的？之后的train中，weight of size 是[2, 1, 3, 3]，需要的depth是1通道的，定位的代码源头是train.py中depth_grad = get_gradient(depth)这条语句，所以哪里可以将depth的通道改变呢？

> pred = np.squeeze(torch.sigmoid(pred)).cpu().data.numpy() filename = os.path.join(r'D:\MyworkSpace\depth\BS-Net-main\output_1', str(i) + '.png') cv2.imwrite(filename, pred) 这样就可以输出深度图 请问一下能得到论文中预期的深度图吗？训练最好的epoch平均loss是多少？方便留个联系方式交流吗？