Commit Graph
6 Commits
Author SHA1 Message Date
Yonghye Kwon 8d1ddc93c7 Earlier assert for cpu and half option (#3508)
* early assert for cpu and half option

early assert for cpu and half option

* Modified comment

Modified comment
2021-06-07 18:56:41 +02:00
Yonghye KwonandGlenn Jocher a1c3572bc9 Explicit model.eval() call if opt.train=False (#3475)
* call model.eval() when opt.train is False

call model.eval() when opt.train is False

* single-line if statement

* cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-06 18:39:36 +02:00
Yonghye Kwon c37f072ba7 Faster HSV augmentation (#3462)
remove datatype conversion process that can be skipped
2021-06-04 20:02:20 +02:00
Yonghye Kwon be86c21c73 rename class autoShape -> AutoShape (#3173)
* rename class autoShape -> AutoShape

follow other class naming convention

* rename class autoShape -> AutoShape

follow other classes' naming convention

* rename class autoShape -> AutoShape
2021-05-16 15:46:45 +02:00
Yonghye Kwon cd8ed3521d image weights compatible faster random index generator v2 for mosaic augmentation (#2383)
image weights compatible faster random index generator v2 for mosaic augmentation
2021-03-05 22:58:26 -08:00
Yonghye KwonandGlenn Jocher 300d518f73 Faster random index generator for mosaic augmentation (#2345)
* 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>
2021-03-05 13:06:18 -08:00