常用的代码段

 prepare.py

这个数据生成代码存在问题。具体请看：神经网络两种数据生成方式训练结果完全不一样_Mick..的博客-CSDN博客

import argparse
import os
import random
import h5py
import numpy as np
import PIL.Image as pil_image
import config
from torchvision import transforms
from data.utils import convert_rgb_to_y
from PIL import Image
import glob
def train(args):
    print('尺度因子是',args.scale)
    if not os.path.exists("../train"):
        os.makedirs("../train")
    h5_file = h5py.File(args.output_path, 'w')
    lr_patches = []
    hr_patches = []
    count = 0
    if args.mode=='y':
        print('生成y通道数据')
    if args.with_aug:
        print('进行数据增强')
    for imag