图像分块 dataset==＞patch

图像分块问题

对于一个大图像的数据集，需要对图像进行分块操作。

class PatchDataset(FnetDataset):

	def __init__(self, dataset, patch_size, buffer_switch_fre=720, npatches=100000, shuffle_images=True):
		self.counter = 0
		self.dataset = dataset
		self.buffer_switch_fre = buffer_switch_fre 
		self.npatches = npatches
		self.shuffle_images = shuffle_images
		self.dim_squeeze = None
		
		# 打乱数据集
		shuffed_data_order = np.arange(0, len(dataset))
		if self.shuffle_images:
            np.random.shuffle(shuffed_data_order)

		# 进度条显示
        pbar = tqdm(range(0, buffer_size))
        self.buffer_history = list(