run train_model.py to start train the model model/model.py contains the definition of used model. model/dataloader contains the different input data structure. measure_img.m and networkvstrad.m are matlab files for testing traditional methods network.m are test for using DNN.
a small test dataset is provided and free to download: https://doi.org/10.6084/m9.figshare.18515108
if you find this work is useful for your research, please cite this paper "Deep learning-based autofocus method enhances image quality in light-sheet fluorescence microscopy"