diff --git a/py_src/yolov4/tf/__init__.py b/py_src/yolov4/tf/__init__.py index 802d7dc9..9366b289 100644 --- 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))