yolov5/utils
Glenn Jocher e54e758cfb
Add PyTorch-only benchmark arg (#7564)
Usage:
```
python utils/benchmarks.py --weights yolov5s.pt --img 640 --pt-only
```
2022-04-24 20:03:23 -07:00
..
aws Fix `yaml.safe_load()` ignore emoji errors (#5060) 2021-10-05 13:41:52 -07:00
docker Add Docker `--file` argument to build (#7527) 2022-04-21 20:31:26 -07:00
flask_rest_api Update restapi.py (#7309) 2022-04-22 16:19:21 -07:00
google_app_engine Bump pip from 19.2 to 21.1 in /utils/google_app_engine (#5661) 2021-11-15 20:10:29 +01:00
loggers Add `--noplots` flag to suppress figures and images logging (#7534) 2022-04-22 15:04:08 -07:00
__init__.py Update bytes to GB with bitshift (#6886) 2022-03-07 19:26:37 +01:00
activations.py [pre-commit.ci] pre-commit suggestions (#7279) 2022-04-04 22:47:00 +02:00
augmentations.py precommit: yapf (#5494) 2022-03-31 16:52:34 +02:00
autoanchor.py Use `tqdm.auto` (#7311) 2022-04-06 12:20:24 +02:00
autobatch.py Update bytes to GB with bitshift (#6886) 2022-03-07 19:26:37 +01:00
benchmarks.py Add PyTorch-only benchmark arg (#7564) 2022-04-24 20:03:23 -07:00
callbacks.py Simplify callbacks.py return (#7333) 2022-04-07 16:15:01 +02:00
datasets.py Disable `pbar` for DDP ranks > 0 (#7440) 2022-04-16 15:12:38 +02:00
downloads.py PyTorch Hub `_verbose=False` fix2 (#7550) 2022-04-23 11:26:25 -07:00
general.py PyTorch Hub `_verbose=False` fix2 (#7550) 2022-04-23 11:26:25 -07:00
loss.py Swap `unsafe_chunk()` for `chunk()` (#7362) 2022-04-10 13:53:28 +02:00
metrics.py Loss and IoU speed improvements (#7361) 2022-04-10 13:46:07 +02:00
plots.py PyTorch Hub `_verbose=False` fix2 (#7550) 2022-04-23 11:26:25 -07:00
torch_utils.py [pre-commit.ci] pre-commit suggestions (#7279) 2022-04-04 22:47:00 +02:00