yolov5/utils
David Strahm d389840f66
Allow int8 quantization for export_tfjs (#10948)
* Allow int8 quantization for export_tfjs

--int8 param currently has no effect on export_tfjs. With this change, 
` python export.py --weights ../path/to/best.pt --include tfjs --int8` will add the --quantize_uint8 param to the tensorflowjs_converter script, greatly reducing model size for web usage.

Signed-off-by: David Strahm <david.strahm@lambda-it.ch>

* Update Dockerfile

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: David Strahm <david.strahm@lambda-it.ch>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-02-10 19:11:08 +04:00
..
aws
docker Allow int8 quantization for export_tfjs (#10948) 2023-02-10 19:11:08 +04:00
flask_rest_api Allow multiple-model serving from Flask REST API (#8973) 2022-08-15 23:42:25 +02:00
google_app_engine [Snyk] Security upgrade gunicorn from 19.9.0 to 19.10.0 (#10933) 2023-02-09 11:52:22 +04:00
loggers Update Dockerfile `FROM pytorch/pytorch:latest` (#10902) 2023-02-06 15:11:32 +04:00
segment Set a seed of generator with an option for more randomness when training several models with different seeds (#10486) 2022-12-19 10:27:34 +01:00
__init__.py Add `join_threads()` (#10086) 2022-11-15 16:27:07 +01:00
activations.py [pre-commit.ci] pre-commit suggestions (#7279) 2022-04-04 22:47:00 +02:00
augmentations.py fix zero labels (#10820) 2023-02-04 16:21:30 +04:00
autoanchor.py New global `TQDM_BAR_FORMAT` (#10211) 2022-11-18 21:39:13 +01:00
autobatch.py AutoBatch report include reserved+allocated (#9491) 2022-09-19 12:18:55 +02:00
callbacks.py Daemon `plot_labels()` for faster start (#9057) 2022-08-21 03:47:37 +02:00
dataloaders.py Security improvements (#10942) 2023-02-09 16:57:18 +04:00
downloads.py Security3 (#10944) 2023-02-09 17:58:24 +04:00
general.py Security3 (#10944) 2023-02-09 17:58:24 +04:00
loss.py Fix torch `long` to `float` tensor on HUB macOS (#8067) 2022-06-01 17:34:46 +02:00
metrics.py Fx confusion-matrix xlabel typo (#10692) 2023-01-14 14:43:27 +01:00
plots.py Revert PIL deprecation fix 2023-01-12 18:01:36 +01:00
torch_utils.py Squeezenet reshape outputs fix (#10222) 2022-11-19 16:44:56 +01:00
triton.py Detect.py supports running against a Triton container (#9228) 2022-09-24 00:56:42 +02:00