python conversion.py
python CityScapes_epistemic_and_aleatoric.py
The model will be saved as model.h5 and if you don't want to train the model again, you can skip this step.
python testResult.py
In the calculation of epistemic uncertainty, the results after dropout of the same image are too similar, resulting in very small variance results, almost no difference, and therefore poor visualization results.
The following are original images, semantic segmentation, aleatoric uncertainty and epistemic uncertainty.