zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

take EXIF orientation tags into account when fixing corrupt images (#5270) 

* take EXIF orientation tags into account when fixing corrupt images

* fit 120 char

* sort imports

* Update local exif_transpose comment

We have a local inplace version that is faster than the official as the image is not copied. AutoShape() uses this for Hub models, but here it is not important as the datasets.py usage is infrequent (AutoShape() it is applied every image).

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
jdfr 2021-10-20 21:35:47 +02:00 committed by GitHub
parent db3bbdde0e
commit 15e8c4c15b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/datasets.py
View File

@ -22,7 +22,7 @@ import numpy as np
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import yaml import yaml
from PIL import Image, ExifTags from PIL import Image, ImageOps, ExifTags
from torch.utils.data import Dataset from torch.utils.data import Dataset
from tqdm import tqdm from tqdm import tqdm
@ -69,7 +69,7 @@ def exif_size(img):
def exif_transpose(image): def exif_transpose(image):
""" """
Transpose a PIL image accordingly if it has an EXIF Orientation tag. Transpose a PIL image accordingly if it has an EXIF Orientation tag.
From https://github.com/python-pillow/Pillow/blob/master/src/PIL/ImageOps.py Inplace version of https://github.com/python-pillow/Pillow/blob/master/src/PIL/ImageOps.py exif_transpose()
:param image: The image to transpose. :param image: The image to transpose.
:return: An image. :return: An image.
@ -896,7 +896,7 @@ def verify_image_label(args):
with open(im_file, 'rb') as f: with open(im_file, 'rb') as f:
f.seek(-2, 2) f.seek(-2, 2)
if f.read() != b'\xff\xd9': # corrupt JPEG if f.read() != b'\xff\xd9': # corrupt JPEG
Image.open(im_file).save(im_file, format='JPEG', subsampling=0, quality=100) # re-save image ImageOps.exif_transpose(Image.open(im_file)).save(im_file, 'JPEG', subsampling=0, quality=100)
msg = f'{prefix}WARNING: {im_file}: corrupt JPEG restored and saved' msg = f'{prefix}WARNING: {im_file}: corrupt JPEG restored and saved'
# verify labels # verify labels