Merge pull request #343 from jadielam/master

Fixing potential bug that can throw IndexError

Author: ifzhang

src/lib/datasets/dataset/jde.py

@@ -443,7 +443,7 @@ def __getitem__(self, files_index):
         bbox_xys = np.zeros((self.max_objs, 4), dtype=np.float32)
 
         draw_gaussian = draw_msra_gaussian if self.opt.mse_loss else draw_umich_gaussian
-        for k in range(num_objs):
+        for k in range(min(num_objs, self.max_objs)):
             label = labels[k]
             bbox = label[2:]
             cls_id = int(label[0])