search

IGNF / myria3d

Myria3D: Aerial Lidar HD Semantic Segmentation with Deep Learning
https://ignf.github.io/myria3d/
BSD 3-Clause "New" or "Revised" License
151 stars 20 forks source link

Add a callback with micro-average Accuracy, Precision, Recall, F1Score and IoU #124

Closed CharlesGaydon closed 2 months ago

CharlesGaydon commented 2 months ago

Log more metrics. Keep the model.py clean. It is hard to unit-test that metrics are actually logged, I'll do some manual checks on known datasets, in mono and multi gpu.

CharlesGaydon commented 2 months ago

Tested on a small patch, with 1 and 2 gpus. Had to add a small related fix in model.py that apparently was not dealt with in the previous bugfix MR.

image

CharlesGaydon commented 2 months ago

Small diff since we are dealing with a single patch + we now use lightning's version instead of a manual computation from a confusion matrix.

CharlesGaydon commented 2 months ago

The diff disapear on larger datasets : image