Oslandia / deeposlandia

Migrated to: https://gitlab.com/Oslandia/deeposlandia
MIT License
59 stars 16 forks source link

Introduce progress bar for inference processes #153

Closed delhomer closed 4 years ago

delhomer commented 4 years ago

This PR makes us using the keras API in order to display progress bar onto the console, when launching inference processes (through inference.py or postprocess.py).

See the following console logging example:

13:11 $ deepo postprocess -D tanzania -b 2 -i grid_034 -s 384
Using TensorFlow backend.
2020-04-30 13:11:48,235 :: INFO :: postprocess :: main : Raw image size: 3850, 3860
2020-04-30 13:11:48,255 :: INFO :: postprocess :: main : The image will be splitted into 121 tiles
2020-04-30 13:11:50,430 :: INFO :: postprocess :: get_trained_model : Model weights have been recovered from ./data/tanzania/output/semseg/checkpoints/best-model-384.h5
2020-04-30 13:11:50,430 :: INFO :: postprocess :: main : Predict labels for input images...
 44/121 [=========>....................] - ETA: 1:29  

Linked to #152.