Improved Code over Baseline for Height Prediction Overview You can train or test the baseline network on DFC2023 dataset. Dependencies Python 3.6 PyTorch 1.7 cuda 11 Usage Stage1 Train Train the baseline network. python train.py Stage2 Test Test the baseline network. python test.py