diff trafficintelligence/moving.py @ 1207:36f0d18e1fad

work in progress on loading kitti (issue with int)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 26 Apr 2023 18:33:29 -0400
parents 3905b393ade0
children 2064e52019db
line wrap: on
line diff
--- a/trafficintelligence/moving.py	Wed Mar 22 23:29:09 2023 -0400
+++ b/trafficintelligence/moving.py	Wed Apr 26 18:33:29 2023 -0400
@@ -4,7 +4,7 @@
 import copy
 from math import sqrt, atan2, cos, sin
 
-from numpy import median, mean, array, arange, zeros, ones, hypot, NaN, std, floor, ceil, float32, argwhere, minimum
+from numpy import median, mean, array, arange, zeros, ones, hypot, NaN, std, floor, ceil, float32, argwhere, minimum,  issubdtype, integer as npinteger
 from matplotlib.pyplot import plot, text
 from scipy.stats import scoreatpercentile
 from scipy.spatial.distance import cdist
@@ -720,7 +720,7 @@
         return self.__len__() == 0
 
     def __getitem__(self, i):
-        if isinstance(i, int):
+        if isinstance(i, int) or issubdtype(i, npinteger):
             return Point(self.positions[0][i], self.positions[1][i])
         elif isinstance(i, slice):
             return Trajectory([self.positions[0][i],self.positions[1][i]])