TangLeDaily/YHRSOD

★ 0Forks 0PythonGitHub ↗Compare

README

YHRSOD

数据集

DUTS-Train

http://saliencydetection.net/duts/download/DUTS-TR.zip

DUTS-Test

http://saliencydetection.net/duts/download/DUTS-TE.zip

  • 在根目录创建以下结构中的datasets\train\ 和 atasets\test\

├── YHRSOD
│    └── datasets
│        ├── train
│            └── DUTS-TR
│        └── test
│            └── DUTS-TE
│    └── rootmodel
│        └── ...
│    └── utils
│        └── ...

  • 将解压后的 DUTS-TR 和 DUTS-TE 放至上述结构位置

训练前配置文件

所有 train_XXX.py 文件中的 parser

训练

python train_RCAN.py 

Contributors

TangLeDaily

Issues