├── .env.sample ├── .gitignore ├── README.md ├── configs ├── callbacks │ └── modelcheckpoint.yaml ├── dataloader │ └── lg_loader.yaml ├── hydra │ └── default.yaml ├── logger │ └── wandb.yaml ├── model │ └── simple_unet.yaml ├── train.yaml └── trainer │ └── default.yaml ├── environment.yaml ├── main.py ├── requirements.txt └── src ├── dataloader └── lg_dataloader.py ├── model ├── simple_unet.py └── utils │ └── psnr.py ├── train.py └── utils.py /.env.sample: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/.env.sample -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/README.md -------------------------------------------------------------------------------- /configs/callbacks/modelcheckpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/callbacks/modelcheckpoint.yaml -------------------------------------------------------------------------------- /configs/dataloader/lg_loader.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/dataloader/lg_loader.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/simple_unet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/model/simple_unet.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/environment.yaml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/dataloader/lg_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/src/dataloader/lg_dataloader.py -------------------------------------------------------------------------------- /src/model/simple_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/src/model/simple_unet.py -------------------------------------------------------------------------------- /src/model/utils/psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/src/model/utils/psnr.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kuielab/lg_dacon_competition_template/HEAD/src/utils.py --------------------------------------------------------------------------------