def resize_label(img, owidth, oheight): img = misc.to_numpy(img) img = cv2.resize( img, (owidth, oheight), 'nearest' ) img = misc.to_torch(img).float() return img
def im_to_torch(img): img = np.transpose(img, (2, 0, 1)) # C*H*W img = misc.to_torch(img).float() return img