Closed dwyzzy closed 4 years ago
自定义语义分割数据集中过滤函数是否有误?
return [img for img in imgs if ( img.size[1] >= self.crop_size[0] and img.size[0] >= self.crop_size[1])]
请问一下此处0、1顺序是否颠倒?
原文相关代码如下:
def filter(self, imgs):
return [img for img in imgs if (
img.shape[0] >= self.crop_size[0] and
img.shape[1] >= self.crop_size[1])]
注意这里使用的是Pytorch推荐的PIL读取image,所以img的size是(w, h)而不是(h, w),例:
from PIL import Image
img = Image.open('xxx.png')
img.size # width, height
注意这里使用的是Pytorch推荐的PIL读取image,所以img的size是(w, h)而不是(h, w),例:
from PIL import Image img = Image.open('xxx.png') img.size # width, height
感谢
自定义语义分割数据集中过滤函数是否有误?
请问一下此处0、1顺序是否颠倒?