From 1a21abde2bbdca0aa36f9d7beba0df284bedb4fd Mon Sep 17 00:00:00 2001
From: daihaojun <‘dhjgm@msn.com’>
Date: Sun, 18 Jun 2023 23:26:27 +0800
Subject: [PATCH] opencv is pretty faster than pillow for image loading

---
 official/cv/YOLOv5/src/initializer.py  | 2 +-
 official/cv/YOLOv5/src/yolo_dataset.py | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/official/cv/YOLOv5/src/initializer.py b/official/cv/YOLOv5/src/initializer.py
index ff65b133c..5d618a2a1 100644
--- a/official/cv/YOLOv5/src/initializer.py
+++ b/official/cv/YOLOv5/src/initializer.py
@@ -16,7 +16,7 @@
 import math
 import mindspore as ms
 from mindspore import nn
-
+from mindspore.train.serialization import load_checkpoint,load_param_into_net
 
 def default_recurisive_init(custom_cell):
     """Initialize parameter."""
diff --git a/official/cv/YOLOv5/src/yolo_dataset.py b/official/cv/YOLOv5/src/yolo_dataset.py
index 00e9dc57c..37d8940f8 100644
--- a/official/cv/YOLOv5/src/yolo_dataset.py
+++ b/official/cv/YOLOv5/src/yolo_dataset.py
@@ -95,8 +95,10 @@ class COCOYoloDataset:
             coco = self.coco
             img_id = self.img_ids[img_ids_index]
             img_path = coco.loadImgs(img_id)[0]["file_name"]
-            img = Image.open(os.path.join(self.root, img_path)).convert("RGB")
-            img = np.array(img)
+            img = cv2.imread(os.path.join(self.root, img_path))
+            img = img[:, :, [2, 1, 0]] #GBR to RGB
+            #img = Image.open(os.path.join(self.root, img_path)).convert("RGB")
+            #img = np.array(img)
             h, w = img.shape[:2]
 
             if i == 0:  # top left
-- 
Gitee