Small error in image_classification.py
VpkPrasanna opened this issue · 1 comments
VpkPrasanna commented
If augmentation is None then we face error as , variable augmented referenced before assignment
UnboundLocalError: local variable 'augmented' referenced before assignment
elif self.backend == "cv2":
image = cv2.imread(self.image_paths[item])
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
if self.resize is not None:
image = cv2.resize(
image,
(self.resize[1], self.resize[0]),
interpolation=cv2.INTER_CUBIC,
)
if self.augmentations is not None:
augmented = self.augmentations(image=image)
image = augmented["image"]
If the indendation is fixed we can solve this error.
abhishekkrthakur commented
fixed