1、运行test.py文件,首先cd进入test.py所在文件夹;设置数据路径,用相对路径可能会混乱,必要时改写成绝对路径;设置检查点路径, ;设置保存路径。
python test.py --data_dir ../dataset/slider_depth --checkpoint_path ../save_dir/2x_7s.pth --save_dir ../save_dir
本机实际输入语句
python test.py --data_dir ../dataset/slider_depth --checkpoint_path /home/guoguangsha/DVS-SR/2x_7s.pth --save_dir ../save_dir
效果界面
最终生成save_dir文件夹,有71张图片
2、运行train.py文件,设置配置路径,选择2x_3.yaml文件;设置数据路径,选择数据路径为dataset/Gray_5K_7s_tiny;设置保存目录save_dir。
python3 train.py --config_path ./configs/2x_3.yaml --data_dir ../dataset/Gray_5K_7s_tiny --save_dir ../save_dir
删除数据集中的.DS_Store文件https://www.javazxz.com/thread-4259-1-1.html
本机实际输入
python3 train.py --config_path ./configs/2x_3.yaml --data_dir ../dataset/dataset/Gray_5K_7s_tiny --save_dir ../save_dir
报错:
Downloading: "https://download.pytorch.org/models/alexnet-owt-7be5be79.pth" to /home/guoguangsha/.cache/torch/hub/checkpoints/alexnet-owt-7be5be79.pth
尝试方法
urllib.error.URLError: <urlopen error [Errno -2] Name or service not known>_是七叔呀的博客-CSDN博客
显存不够,把batch_size=32改为batch_size=16,运行结果如下: