Depth estimation for SEM images

Samsung project, manufacturing AI, 2022 Spring

Requirements

conda env create -f conda_env.yml
conda activate samsung

Instructions

Dataset is required in data/.

Train

To run training use the train.py script

python train.py

for 4-to-1 depth estimation, or

python train.py --single_image

for 1-to-1 depth estimation.

Predict

python predict.py

for 4-to-1 depth estimation, or

python predict_one.py

for 1-to-1 depth estimation.