AceCoooool / DSS-pytorch

:star: PyTorch implement of Deeply Supervised Salient Object Detection with Short Connection
MIT License
174 stars 53 forks source link

how to get MSRA-B file construction? #20

Open panmyuan opened 5 years ago

AceCoooool commented 5 years ago

please see #9 ,and if you want to split by yourself. Follow the step:

  1. read all image name (5000 image name) --- can use os.listdir
  2. shuffle the 0-4999 index, and choose 2500 as train, 500 as valid, 2000 as test. save image name to txt file.
panmyuan commented 5 years ago

OK,thank you