(self,dataset,size,h,w)
| 215 | self.real_H = GT_IMG.to(self.device) |
| 216 | |
| 217 | def split2(self,dataset,size,h,w): |
| 218 | newdataset=[] |
| 219 | nsize1=256 |
| 220 | nsize2=256 |
| 221 | for i in range (size): |
| 222 | im=dataset[i] |
| 223 | for ii in range(0,h,nsize1): #2048 |
| 224 | for iii in range(0,w,nsize2): #1536 |
| 225 | newdataset.append(im[ii:ii+nsize1,iii:iii+nsize2,:]) |
| 226 | |
| 227 | return np.array(newdataset) |
| 228 | def merge_image2(self,splitted_images, h,w): |
| 229 | image=np.zeros(((h,w,3))) |
| 230 | nsize1=256 |