pytorch学习
qq_23304241
这个作者很懒,什么都没留下…
展开
-
pytorch测试的时候为何要加上model.eval()
Do need to use model.eval() when I test?Sure, Dropout works as a regularization for preventing overfitting during training.It randomly zeros the elements of inputs in Dropout layer on forward call...原创 2019-08-14 19:59:21 · 13109 阅读 · 5 评论 -
pytorch只用中要注意通道问题
cv读进来的是BGR图像,通道是hcw,在torch中使用要注意维度转换def __getitem__(self, idx): '''Load image. Args: idx: (int) image index. img_org = Image.open(self.root_src + 'reference_cutB...原创 2019-08-14 20:08:23 · 3149 阅读 · 1 评论 -
Pytorch中通道转化问题
cv读进来的是BGR图像,通道是hcw,在torch中使用要注意维度转换def __getitem__(self, idx): '''Load image. Args: idx: (int) image index. img_org = Image.open(self.root_src + 'reference_cutB...原创 2019-08-14 20:19:56 · 8314 阅读 · 1 评论