- pytorch
- opencv
- sklearn
- skimage
- tqdm
Down the checkpoint of correction net from Checkpoints, create a new folder named MegOCT/weights/ and place the checkpoint in it.
We have prepared some demonstration data and you can down them from. Then just place the data in this folder.
Just run python oct_restore.py to start inferences. By default, the outputs are saved in ./outputs.