Update autoanchor.py

ultralytics · Feb 5, 2021 · 4450fb4 · 4450fb4
1 parent 5742a78
commit 4450fb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/autoanchor.py b/utils/autoanchor.py
@@ -100,7 +100,7 @@ def print_results(k):
         with open(path) as f:
             data_dict = yaml.load(f, Loader=yaml.SafeLoader)  # model dict
         from utils.datasets import LoadImagesAndLabels
-        dataset = LoadImagesAndLabels(data_dict['train'], data_type="train", augment=True, rect=True)
+        dataset = LoadImagesAndLabels(data_dict['train'], augment=True, rect=True)
     else:
         dataset = path  # dataset