fitsdata/
fitsdata/fits_train
fitsdata/fits_train/587722983904051292/587722983904051292-g.fits
fitsdata/fits_train/587722983904051292/587722983904051292-i.fits
fitsdata/fits_train/587722983904051292/587722983904051292-r.fits
fitsdata/fits_train/587722983904051292/587722983904051292-u.fits
fitsdata/fits_train/587722983904051292/587722983904051292-z.fits
...
fitsdata/fits_test
fitsdata/fits_test/587722981744771128/587722981744771128-g.fits
fitsdata/fits_test/587722981744771128/587722981744771128-i.fits
fitsdata/fits_test/587722981744771128/587722981744771128-r.fits
fitsdata/fits_test/587722981744771128/587722981744771128-u.fits
fitsdata/fits_test/587722981744771128/587722981744771128-z.fits
...
python fits2jpg.py --input your_fits_data_folder --mode 0 #train data mode
python fits2jpg.py --input your_fits-data_foleer --mode 1 #test data mode