tf: fix strange ground truth values issue

close: #38
hhk7734 · Dec 5, 2020 · 22c12fa · 22c12fa
1 parent 483443f
commit 22c12fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py_src/yolov4/tf/__init__.py b/py_src/yolov4/tf/__init__.py
@@ -317,7 +317,7 @@ def save_dataset_for_mAP(
 
         for i in range(num_sample):
             # image_path, [[x, y, w, h, class_id], ...]
-            _dataset = data_set.dataset[i % max_dataset_size]
+            _dataset = data_set.dataset[i % max_dataset_size].copy()
 
             if images_optional:
                 image_path = path.join(img_dir_path, "image_{}.jpg".format(i))