Ultrasound fetal images

Description

What it does

Install dependencies with Makefile

make install

Active conda environment

conda activate ultrasound_fetal_images_env

Train model with default configuration

# train on CPU
python src/train.py trainer=cpu

# train on GPU
python src/train.py trainer=gpu

Train model with chosen experiment configuration from configs/experiment/

python src/train.py experiment=experiment_name.yaml

You can override any parameter from command line like this

python src/train.py trainer.max_epochs=20 data.batch_size=64

Name		Name	Last commit message	Last commit date
Latest commit History 119 Commits
.github		.github
configs		configs
data		data
hydra_plugins		hydra_plugins
logs		logs
notebooks		notebooks
scripts		scripts
src		src
tests		tests
.env.example		.env.example
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
.project-root		.project-root
Makefile		Makefile
README.md		README.md
conda-linux-64.lock		conda-linux-64.lock
environment.yml		environment.yml
poetry.lock		poetry.lock
poetry.toml		poetry.toml
pyproject.toml		pyproject.toml
setup.py		setup.py
virtual-packages.yml		virtual-packages.yml