def get_edges(self, idx, rot=0, zoom=1.0):
mask_d_img = DepthImage.from_tiff(self.mask_df[idx])
mask_d_img.rotate(rot)
mask_d_img.zoom(zoom)
mask_d_img.resize((self.output_size, self.output_size))
edge_img = cv2.Canny(mask_d_img.img, 20, 30)
edge_img = (edge_img / 255).astype(np.uint8)
return edge_img
def get_mask_d_1(self,idx,rot = 0,zoom = 1.0):
'''
:功能 :读取返回指定id的depth图像
:参数 idx :int,要读取的数据id
:返回 :ndarray,处理好后的depth图像
'''
mask_d_img = DepthImage.from_tiff(self.mask_df1[idx])
mask_d_img.rotate(rot)
mask_d_img.zoom(zoom)
mask_d_img.resize((self.output_size,self.output_size))
return mask_d_img.img
def get_depth(self, idx, rot=0, zoom=1.0):
depth_img = DepthImage.from_file(self.depthf[idx])
center, left, top = self._get_crop_attrs(idx)
#先旋转后裁剪再缩放最后resize
depth_img.rotate(rot, center)
depth_img.crop((top, left), (min(
480, top + self.output_size), min(640, left + self.output_size)))
depth_img.normalize()
depth_img.zoom(zoom)
depth_img.resize((self.output_size, self.output_size))
return depth_img.img
def get_depth(self,idx,rot = 0,zoom = 1.0):
'''
:功能 :读取返回指定id的depth图像
:参数 idx :int,要读取的数据id
:返回 :ndarray,处理好后的depth图像
'''
depth_img = DepthImage.from_tiff(self.depthf[idx])
depth_img.rotate(rot)
depth_img.normalize()
depth_img.zoom(zoom)
depth_img.resize((self.output_size,self.output_size))
return depth_img.img
def get_mask_prob(self,idx,rot = 0,zoom = 1.0):
'''
:功能 :读取返回指定id的depth图像
:参数 idx :int,要读取的数据id
:返回 :ndarray,处理好后的depth图像
'''
mask_prob_img = DepthImage.from_tiff(self.mask_prob[idx])
# mask_prob_img 除以其最大值,来将值缩放到[0,1]范围内
img_max = mask_prob_img.img.max()
mask_prob_img.rotate(rot)
mask_prob_img.zoom(zoom)
mask_prob_img.resize((self.output_size,self.output_size))
return mask_prob_img.img / img_max
def get_depth(self, idx):
'''
:功能 :读取返回指定id的depth图像
:参数 idx :int,要读取的数据id
:返回 :ndarray,处理好后的depth图像
'''
#目前这个DepthImage类还没有定义，后面仿照Image类给它定义一下
depth_img = DepthImage.from_file(self.depthf[idx])
depth_img.normalize()
center, left, top = self._get_crop_attrs(idx)
depth_img.crop((left, top),
(left + self.output_size, top + self.output_size))
depth_img.resize((self.output_size, self.output_size))
return depth_img.img
def get_depth(self,idx,rot=0, zoom=1.0):
'''
:功能 :读取返回指定id的depth图像
:参数 idx :int,要读取的数据id
:返回 :ndarray,处理好后的depth图像
'''
#目前这个DepthImage类还没有定义，后面仿照Image类给它定义一下
depth_img = DepthImage.from_file(self.depthf[idx])
center,left,top = self._get_crop_attrs(idx)
#先旋转后裁剪再缩放最后resize
depth_img.rotate(rot,center)
depth_img.crop((top,left),(min(480,top+self.output_size),min(640,left+self.output_size)))
depth_img.normalize()
depth_img.zoom(zoom)
depth_img.resize((self.output_size, self.output_size))
return depth_img.img