Optimize UNet with Optuna on Copick Data

A solution that optimizes a 3D UNet model using Optuna, with data from Copick.

Arguments

--copick_config_path

Path to the Copick configuration JSON file. (default value: PARAMETER_VALUE)

--voxel_spacing

Voxel spacing used to scale pick locations. (default value: PARAMETER_VALUE)

--tomo_type

Tomogram type to use for each tomogram. (default value: PARAMETER_VALUE)

--epochs

Number of training epochs. (default value: PARAMETER_VALUE)

--random_seed

Random seed for reproducibility. (default value: 17171)

Please follow this link for details on how to install and run this solution.