yolov5/utils
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
..
aws Update resume.py (#4115) 2021-07-22 17:22:11 +02:00
flask_rest_api Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
google_app_engine Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 2021-06-09 21:14:09 +02:00
loggers W&B: Fix sweep bug (#4276) 2021-08-02 18:41:52 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py ACON Activation batch-size 1 bug patch (#2901) 2021-04-25 22:54:07 +02:00
augmentations.py Albumentations >= 1.0.3 (#4068) 2021-07-19 13:23:19 +02:00
autoanchor.py Update script headers (#4163) 2021-07-26 15:23:33 +02:00
callbacks.py Add `train.py` and `val.py` callbacks (#4220) 2021-08-01 00:18:07 +02:00
datasets.py Feature `python train.py --cache disk` (#4049) 2021-08-02 18:47:24 +02:00
downloads.py PyCharm reformat (#4209) 2021-07-28 23:35:14 +02:00
general.py Add `train.py` and `val.py` callbacks (#4220) 2021-08-01 00:18:07 +02:00
loss.py New CSV Logger (#4148) 2021-07-25 19:06:37 +02:00
metrics.py Copy-Paste augmentation for YOLOv5 (#3845) 2021-07-01 00:35:04 +02:00
plots.py Add `train.py` and `val.py` callbacks (#4220) 2021-08-01 00:18:07 +02:00
torch_utils.py Update `profile()` for CUDA Memory allocation (#4239) 2021-07-30 18:17:19 +02:00