上传者: 38649657
|
上传时间: 2021-05-26 18:03:08
|
文件大小: 12KB
|
文件类型: ZIP
预算matlab代码DVTSR
PyTorch实施“”
相关工作
PyTorch实现上的,sRGB轨道的第二名和Raw-RGB轨道的第三名。
()
PyTorch重新实现了Simon
Niklaus的“”。
()
依存关系
Python
3.6
PyTorch
1.0.0
的MATLAB
TensorFlow
张量板TensorBoardX
火炬摘要
数据
我们使用VTSR数据集进行训练。
要生成训练数据,请使用Matlab代码generate_train.m
训练
例如,使用以下命令来使用我们的培训代码
python
main_tsr.py
--lr
1e-4
--step
2
--cuda
True
--train_data0
./train_data0.h5
--train_data1
./train_data1.h5
--train_label
./train_label.h5
--valid_data0
./valid_data0.h5
--valid_data1
./valid_data1.h5
--valid_label
./valid_label.h5
--gpu
0,