def load_data(self, path, resize_height, resize_width, normalization):
'''
加载数据
:param path:
:param resize_height:
:param resize_width:
:param normalization: 是否归一化
:return:
'''
image = image_processing.read_image(path, resize_height, resize_width, normalization)
return image
def select_user_roi(image_path):
'''
由于原图的分辨率较大，这里缩小后获取ROI，返回时需要重新scale对应原图
:param image_path:
:return:
'''
orig_image = image_processing.read_image(image_path)
orig_shape = np.shape(orig_image)
resize_image = image_processing.resize_image(orig_image,
resize_height=800,
resize_width=None)
re_shape = np.shape(resize_image)
g_rect = get_image_roi(resize_image)
orgi_rect = image_processing.scale_rect(g_rect, re_shape, orig_shape)
roi_image = image_processing.get_rect_image(orig_image, orgi_rect)
image_processing.cv_show_image("RECT", roi_image)
image_processing.show_image_rect("image", orig_image, orgi_rect)
return orgi_rect