Fix visualization of unsigned integer attributes

Author: tandede

ml3d/vis/visualizer.py

@@ -141,7 +141,7 @@ def _convert_to_numpy(self, ary):
         elif isinstance(ary, np.ndarray):
             if len(ary.shape) == 2 and ary.shape[0] == 1:
                 ary = ary[0]  # "1D" array as 2D: [[1, 2, 3,...]]
-            if ary.dtype.name.startswith('int'):
+            if np.issubdtype(ary.dtype, np.integer):
                 return np.array(ary, dtype='float32')
             else:
                 return ary

tests/test_visualizer.py

@@ -0,0 +1,14 @@
+import numpy as np
+import pytest
+
+from ml3d.vis.visualizer import Model
+
+
+@pytest.mark.parametrize("dtype", [np.uint8, np.uint16, np.uint32, np.uint64])
+def test_unsigned_integer_attributes_are_converted_to_float32(dtype):
+    values = np.array([0, 1, 2], dtype=dtype)
+
+    converted = Model()._convert_to_numpy(values)
+
+    assert converted.dtype == np.float32
+    np.testing.assert_array_equal(converted, values)