import?argparse
import?glob
from?PIL?import?Image
import?PIL
import?random
from?utils?import?*

#?the?pixel?value?range?is?‘0-255‘（uint8?）?of?training?data

#?macro
DATA_AUG_TIMES?=?1??#?transform?a?sample?to?a?different?sample?for?DATA_AUG_TIMES?times

parser?=?argparse.ArgumentParser（description=‘‘）
parser.add_argument（‘--src_dir‘?dest=‘src_dir‘?default=‘./data/Train400‘?help=‘dir?of?data‘）
parser.add_argument（‘--save_dir‘?dest=‘save_dir‘?default=‘./data‘?help=‘dir?of?patches‘）
parser.add_argument（‘--patch_size‘?dest=‘pat_size‘?type=int?default=40?help=‘patch?size‘）
parser.add_argument（‘--stride‘?dest=‘stride‘?type=int?default=10?help=‘stride‘）
parser.add_argument（‘--step‘?dest=‘step‘?type=int?default=0?help=‘step‘）
parser.add_argument（‘--batch_size‘?dest=‘bat_size‘?type=int?default=128?help=‘batch?size‘）
#?check?output?arguments
parser.add_argument（‘--from_file‘?dest=‘from_file‘?default=“./data/img_clean_pats.npy“?help=‘get?pic?from?file‘）
parser.add_argument（‘--num_pic‘?dest=‘num_pic‘?type=int?default=10?help=‘number?of?pic?to?pick‘）
args?=?parser.parse_args（）


def?generate_patches（isDebug=False）:
????global?DATA_AUG_TIMES
????count?=?0
????filepaths?=?glob.glob（args.src_dir?+?‘/*.png‘）
????if?isDebug:
????????filepaths?=?filepaths[:10]
????print（“number?of?training?data?%d“?%?len（filepaths））
????
????scales?=?[1?0.9?0.8?0.7]
????
????#?calculate?the?number?of?patches
????for?i?in?range（len（filepaths））:
????????img?=?Image.open（filepaths[i]）.convert（‘L‘）??#?convert?RGB?to?gray
????????for?s?in?range（len（scales））:
????????????newsize?=?（int（img.size[0]?*?scales[s]）?int（img.size[1]?*?scales[s]））
????????????img_s?=?img.resize（newsize?resample=PIL.Image.BICUBIC）??#?do?not?change?the?original?img
????????????im_h?im_w?=?img_s.size
????????????for?x?in?range（0?+?args.step?（im_h?-?args.pat_size）?args.stride）:
????????????????for?y?in?range（0?+?args.step?（im_w?-?args.pat_size）?args.stride）:
????????????????????count?+=?1
????origin_patch_num?=?count?*?DATA_AUG_TIMES
????
????if?origin_patch_num?%?args.bat_size?!=?0:
????????numPatches?=?（origin_patch_num?/?args.bat_size?+?1）?*?args.bat_size
????else:
????????numPatches?=?origin_patch_num
????print（“total?patches?=?%d??batch?size?=?%d?total?batches?=?%d“?%?\
??????????（numPatches?args.bat_size?numPatches?/?args.bat_size））
????
????#?data?matrix?4-D
????inputs?=?np.zeros（（numPatches?args.pat_size?args.pat_size?1）?dtype=“uint8“）
????
????count?=?0
????#?generate?patches
????for?i?in?range（len（filepaths））:
????????img?=?Image.open（filepaths[i]）.convert（‘L‘）
????????for?s?in?range（len（scales））:
????????????newsize?=?（int（img.size[0]?*?scales[s]）?int（img.size[1]?*?scales[s]））
????????????#?print?newsize
????????????img_s?=?img.resize（newsize?resample=PIL.Image.BICUBIC）
????????????img_s?=?np.reshape（np.array（img_s?dtype=“uint8“）
???????????????????????????????（img_s.size[0]?img_s.size[1]

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-11-02?14:47??DnCNN\
?????文件??????????67??2018-01-12?10:30??DnCNN\.gitignore
?????目錄???????????0??2018-11-02?14:48??DnCNN\.idea\
?????文件?????????459??2018-11-02?14:38??DnCNN\.idea\DnCNN.iml
?????目錄???????????0??2018-11-02?14:37??DnCNN\.idea\inspectionProfiles\
?????文件?????????228??2018-11-02?14:37??DnCNN\.idea\inspectionProfiles\profiles_settings.xml
?????文件?????????219??2018-11-02?14:37??DnCNN\.idea\misc.xml
?????文件?????????262??2018-11-02?14:37??DnCNN\.idea\modules.xml
?????文件???????19606??2018-11-02?14:48??DnCNN\.idea\workspace.xml
?????文件?????????195??2018-01-12?10:30??DnCNN\CONTRIBUTING.md
?????文件?????????186??2018-01-12?10:30??DnCNN\Dockerfile
?????文件???????35141??2018-01-12?10:30??DnCNN\LICENSE
?????文件????????3308??2018-01-12?10:30??DnCNN\README.md
?????目錄???????????0??2018-11-02?14:41??DnCNN\__pycache__\
?????文件????????6037??2018-11-02?14:41??DnCNN\__pycache__\model.cpython-36.pyc
?????文件????????2946??2018-11-02?14:41??DnCNN\__pycache__\utils.cpython-36.pyc
?????目錄???????????0??2018-11-02?14:49??DnCNN\checkpoint\
?????目錄???????????0??2018-11-02?14:37??DnCNN\checkpoint_demo\
?????文件?????6680852??2018-01-12?10:30??DnCNN\checkpoint_demo\DnCNN-tensorflow-67336.data-00000-of-00001
?????文件????????7095??2018-01-12?10:30??DnCNN\checkpoint_demo\DnCNN-tensorflow-67336.index
?????文件?????3924539??2018-01-12?10:30??DnCNN\checkpoint_demo\DnCNN-tensorflow-67336.meta
?????文件?????????101??2018-01-12?10:30??DnCNN\checkpoint_demo\checkpoint
?????目錄???????????0??2018-11-02?14:37??DnCNN\data\
?????目錄???????????0??2018-11-02?14:37??DnCNN\data\Train400\
?????文件???????20790??2018-01-12?10:30??DnCNN\data\Train400\test_001.png
?????文件???????14758??2018-01-12?10:30??DnCNN\data\Train400\test_002.png
?????文件???????22658??2018-01-12?10:30??DnCNN\data\Train400\test_003.png
?????文件???????20250??2018-01-12?10:30??DnCNN\data\Train400\test_004.png
?????文件???????20048??2018-01-12?10:30??DnCNN\data\Train400\test_005.png
?????文件???????18456??2018-01-12?10:30??DnCNN\data\Train400\test_006.png
?????文件???????19182??2018-01-12?10:30??DnCNN\data\Train400\test_007.png
............此處省略422個文件信息