From 9753777cafac1cb177a8a3602e09850a98663708 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Wed, 16 Jun 2021 11:47:10 +0000 Subject: [PATCH] fix vehicle config --- ppcls/configs/Vehicle/ResNet50.yaml | 6 ------ ppcls/data/dataloader/vehicle_dataset.py | 4 ++-- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/ppcls/configs/Vehicle/ResNet50.yaml b/ppcls/configs/Vehicle/ResNet50.yaml index e8741798..927a975e 100644 --- a/ppcls/configs/Vehicle/ResNet50.yaml +++ b/ppcls/configs/Vehicle/ResNet50.yaml @@ -73,9 +73,6 @@ DataLoader: bbox_crop: True cls_label_path: "./dataset/CompCars/train_test_split/classification/train_label.txt" transform_ops: - - DecodeImage: - to_rgb: True - channel_first: False - ResizeImage: size: 224 - RandFlipImage: @@ -113,9 +110,6 @@ DataLoader: cls_label_path: "./dataset/CompCars/train_test_split/classification/test_label.txt" bbox_crop: True transform_ops: - - DecodeImage: - to_rgb: True - channel_first: False - ResizeImage: size: 224 - NormalizeImage: diff --git a/ppcls/data/dataloader/vehicle_dataset.py b/ppcls/data/dataloader/vehicle_dataset.py index eb687aea..baae63c2 100644 --- a/ppcls/data/dataloader/vehicle_dataset.py +++ b/ppcls/data/dataloader/vehicle_dataset.py @@ -117,8 +117,8 @@ class VeriWild(Dataset): def __getitem__(self, idx): try: - img = cv2.imread(self.images[idx]) - img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) + with open(self.images[idx], 'rb') as f: + img = f.read() if self._transform_ops: img = transform(img, self._transform_ops) img = img.transpose((2, 0, 1)) -- GitLab