fix #120 #184

lufficc · Nov 13, 2020 · 68dc0a2 · 68dc0a2
1 parent 50373c7
commit 68dc0a2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ssd/data/build.py b/ssd/data/build.py
@@ -33,7 +33,7 @@ def make_data_loader(cfg, is_train=True, distributed=False, max_iter=None, start
     dataset_list = cfg.DATASETS.TRAIN if is_train else cfg.DATASETS.TEST
     datasets = build_dataset(dataset_list, transform=train_transform, target_transform=target_transform, is_train=is_train)
 
-    shuffle = is_train or distributed
+    shuffle = is_train
 
     data_loaders = []
 

diff --git a/ssd/engine/inference.py b/ssd/engine/inference.py
@@ -44,7 +44,7 @@ def compute_on_dataset(model, data_loader, device):
 
             outputs = [o.to(cpu_device) for o in outputs]
         results_dict.update(
-            {img_id: result for img_id, result in zip(image_ids, outputs)}
+            {int(img_id): result for img_id, result in zip(image_ids, outputs)}
         )
     return results_dict