t

openvinotoolkit · Nov 15, 2024 · c93b019 · c93b019
1 parent 63fdf19
commit c93b019
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 11 deletions.
diff --git a/examples/post_training_quantization/onnx/yolov8_quantize_with_accuracy_control/main.py b/examples/post_training_quantization/onnx/yolov8_quantize_with_accuracy_control/main.py
@@ -30,6 +30,8 @@
 import nncf
 from nncf.common.logging.track_progress import track
 
+MODEL_NAME = "yolov8n-seg"
+
 ROOT = Path(__file__).parent.resolve()
 DATASET_PATH = Path().home() / ".cache" / "nncf" / "datasets"
 
@@ -110,6 +112,10 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[SegmentationValidator, t
     validator: SegmentationValidator = model.task_map[model.task]["validator"](args=args)
     validator.data = check_det_dataset(args.data)
     validator.stride = 32
+
+    coco_data_path = DATASET_PATH / "coco128-seg"
+    data_loader = validator.get_dataloader(coco_data_path.as_posix(), 1)
+
     validator.is_coco = True
     validator.class_map = coco80_to_coco91_class()
     validator.names = model.model.names
@@ -118,9 +124,6 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[SegmentationValidator, t
     validator.process = ops.process_mask
     validator.plot_masks = []
 
-    coco_data_path = DATASET_PATH / "coco128-seg"
-    data_loader = validator.get_dataloader(coco_data_path.as_posix(), 1)
-
     return validator, data_loader
 
 
@@ -218,8 +221,6 @@ def validation_ac(
 
 
 def run_example():
-    MODEL_NAME = "yolov8n-seg"
-
     model = YOLO(ROOT / f"{MODEL_NAME}.pt")
     args = get_cfg(cfg=DEFAULT_CFG)
     args.data = "coco128-seg.yaml"

diff --git a/examples/post_training_quantization/openvino/yolov8/main.py b/examples/post_training_quantization/openvino/yolov8/main.py
@@ -75,15 +75,16 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[DetectionValidator, torc
     validator: DetectionValidator = model.task_map[model.task]["validator"](args=args)
     validator.data = check_det_dataset(args.data)
     validator.stride = 32
+
+    coco_data_path = DATASET_PATH / "coco128"
+    data_loader = validator.get_dataloader(coco_data_path.as_posix(), 1)
+
     validator.is_coco = True
     validator.class_map = coco80_to_coco91_class()
     validator.names = model.model.names
     validator.metrics.names = validator.names
     validator.nc = model.model.model[-1].nc
 
-    coco_data_path = DATASET_PATH / "coco128"
-    data_loader = validator.get_dataloader(coco_data_path.as_posix(), 1)
-
     return validator, data_loader
 
 

diff --git a/examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/main.py b/examples/post_training_quantization/openvino/yolov8_quantize_with_accuracy_control/main.py
@@ -100,6 +100,10 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[SegmentationValidator, t
     validator: SegmentationValidator = model.task_map[model.task]["validator"](args=args)
     validator.data = check_det_dataset(args.data)
     validator.stride = 32
+
+    coco_data_path = DATASET_PATH / "coco128-seg"
+    data_loader = validator.get_dataloader(coco_data_path.as_posix(), 1)
+
     validator.is_coco = True
     validator.class_map = coco80_to_coco91_class()
     validator.names = model.model.names
@@ -108,9 +112,6 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[SegmentationValidator, t
     validator.process = ops.process_mask
     validator.plot_masks = []
 
-    coco_data_path = DATASET_PATH / "coco128-seg"
-    data_loader = validator.get_dataloader(coco_data_path.as_posix(), 1)
-
     return validator, data_loader