2022年3月1日
pytorch继承nn.Module类定义模型
摘要： pytorch继承nn.Module类定义模型 在pytorch中，最常用于定义模型的方法是继承nn.Module类后重载__init__()和forward函数。部分疑问记录： 1.为什么重载forward函数后可以直接使用net(x)调用？ 2.哪些网络模块要预先写在__init__中？ 3.如 阅读全文
posted @ 2022-03-01 18:11 _海阔天空 阅读(678) 评论(0) 推荐(0)
区别 image.shape[0],image.shape[1],image.shape[2]
摘要： 区别 image.shape[0],image.shape[1],image.shape[2] image.shape[0], 图片垂直尺寸 image.shape[1], 图片水平尺寸 image.shape[2], 图片通道数 python的img.shape函数 - 走看看 (zoukanka 阅读全文
posted @ 2022-03-01 13:00 _海阔天空 阅读(1254) 评论(0) 推荐(0)