Fix type error in DropPointsByClass

IGNF / myria3d

Myria3D: Aerial Lidar HD Semantic Segmentation with Deep Learning

https://ignf.github.io/myria3d/

BSD 3-Clause "New" or "Revised" License

172 stars 23 forks source link

Fix type error in DropPointsByClass #113

Closed leavauchier closed 7 months ago

leavauchier commented 7 months ago

A TypeError: object of type 'numpy.int64' has no len() was raised in some specific use case when using DropPointsByClass with batches containing a single point (the numpy array was transformed to an integer)

CharlesGaydon commented 7 months ago

LGTM !