上传者: 38600696
|
上传时间: 2021-12-17 16:32:53
|
文件大小: 583.59MB
|
文件类型: -
订单分批matlab代码神经PGD
该代码实现了中提出的神经近端梯度下降(PGD)算法。
想法是展开近端梯度下降算法,并使用神经网络对近端进行建模。
采用残差网络(ResNet)作为近端,实现递归神经网络(RNN)学习近端。
该代码可以灵活地结合各种培训成本,包括:1)逐像素l1
/
l2、2)SSIM和3)对抗性GAN,LSGAN和WGAN。
命令行
python3
npgd_main.py
-火车--dataset_train
/路径/到/火车/数据集--dataset_test
/路径/到/测试/数据集--sampling_pattern
/路径/到/采样/轨迹/.matfile
--sample_size_x
320
--sample_size_y
256
--batch_size
2
--summary_period
20000
--subsample_test
1000
--subsample_train
1000
--train_time
3000
--train_dir
/路径/到/保存/结果--checkpoint_dir
/路径/到/保存/检查点--tensorbo