8 Commits

Author SHA1 Message Date
junji hashimoto
2d99063201
Feature python train.py --cache disk (#4049)
* Add cache-on-disk and cache-directory to cache images on disk

* Fix load_image with cache_on_disk

* Add no_cache flag for load_image

* Revert the parts('logging' and a new line) that do not need to be modified

* Add the assertion for shapes of cached images

* Add a suffix string for cached images

* Fix boundary-error of letterbox for load_mosaic

* Add prefix as cache-key of cache-on-disk

* Update cache-function on disk

* Add psutil in requirements.txt

* Update train.py

* Cleanup1

* Cleanup2

* Skip existing npy

* Include re-space

* Export return character fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-08-02 18:47:24 +02:00
Glenn Jocher
388016e9e3
Update 'results saved to' string (#4275) 2021-08-02 15:48:53 +02:00
Glenn Jocher
6e4358f3f3
Add export.py ONNX inference suggestion (#4146) 2021-07-25 13:46:16 +02:00
Glenn Jocher
264be1a616
Rename opset_version to opset (#4135) 2021-07-24 13:19:12 +02:00
Glenn Jocher
3bef77f5cb
Addition refactor export.py (#4089)
* Addition refactor `export.py`

* Update export.py
2021-07-20 18:42:27 +02:00
Glenn Jocher
442a7abdf2
Refactor export.py (#4080)
* Refactor `export.py`

* cleanup

* Update check_requirements()

* Update export.py
2021-07-20 13:21:52 +02:00
fcakyon
0e2d0d54d7
Fix typo (#3729) 2021-06-22 13:33:38 +02:00
Glenn Jocher
1f69d12591
Update 4 main ops for paths and .run() (#3715)
* Add yolov5/ to path

* rename functions to run()

* cleanup

* rename fix

* CI fix

* cleanup find models/export.py
2021-06-21 17:25:04 +02:00