6 from .base_dataloader
import _BaseImageLoader
7 from ..factory
import DATALOADERS
14 self.
_to_rgb = kwargs.pop(
'toRGB',
False)
17 def _toRGB(self, image):
18 return cv.cvtColor(image, cv.COLOR_BGR2RGB)
20 def _centerCrop(self, image):
26 return image[w_crop:w-w_crop, h_crop:h-h_crop, :]
29 for filename
in self.
_files:
30 image = cv.imread(os.path.join(self.
_path, filename))
39 image = cv.resize(image, size)