Faster random index generator for mosaic augmentation (#2345) · fhkiel-mlaip/yolov5@e3a9228 · GitHub

Commit

Faster random index generator for mosaic augmentation (ultralytics#2345)

Browse files

* faster random index generator for mosaic augementation

We don't need to access list to generate random index

It makes augmentation slower.

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Loading branch information

developer0hye and glenn-jocher committed Mar 5, 2021

1 parent 05895e9 commit e3a9228