This repository does not have data/model_unet folder and has only inference script in notebook 4_ml_coverage_map.ipynb with pre-trained models. Please share the pre-trained models -
- 1st_checkpoint_epoch58.pth and
- 2nd_checkpoint_epoch110.pth
Also , both UNET training code is not present. Pls.add to the repository