Commit Graph

159 Commits (52d96222bbfa1b1b5a6d4f1086ae9d07b914879f)

Author SHA1 Message Date
Glenn Jocher 57c342a6a5
Simplify ONNX inference command (#4405) 2021-08-13 13:22:13 +02:00
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