tensorflow-yolov4 icon indicating copy to clipboard operation
tensorflow-yolov4 copied to clipboard

Increase YOLODataset speed

Open ccasadei opened this issue 3 years ago • 0 comments

Hi,

using TurboJpeg instead of cv2, I increased the dataset image reading speed from 80 images/sec to 126 images/sec on my pc.

Method: _convert_dataset_to_image_and_bboxes

try:
    # image = cv2.imread(dataset[0])
    # image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    with open(dataset[0], "rb") as f:
        image = self.jpgdecode.decode(f.read(), pixel_format=TJPF_RGB)
except:
    return None, None

where self.jpgdecode is assigned once in __init__ constructor:

...
    self.jpgdecode = turbojpeg.TurboJPEG()
...

ccasadei avatar Jun 01 '21 07:06 ccasadei