git clone https://github.com/backseason/PoolNet.git
cd PoolNet/
Download the following pre-trained models GoogleDrive | BaiduYun (pwd: 27p5) into checkpoint
folder.
For single dataset testing: *
changes accordingly and --sal_mode
indicates different datasets (details can be found in main.py
)
python create_bop.py