Glenn Jocher
a64a4c839f
Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()` ( #4919 )
* Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()`
* Cleanup
3 years ago
Jebastin Nadar
1492632796
Fix zipfile name for coco128-segments ( #4914 )
3 years ago
Jiacong Fang
2c2ef25f8b
TensorFlow.js export enhancements ( #4905 )
* Add arguments to TensorFlow NMS call
* Add regex substitution to reorder Identity_*
* Delete reorder in docstring
* Cleanup
* Cleanup2
* Removed `+ \` on string ends (not needed)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Jiacong Fang
ce7fa81d4e
Avoid out-of-image class labels ( #4902 )
* Avoid out-of-image class labels
* Update plots.py
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
NauchtanRobotics
dad8660540
Fix arg help string to match 'classes' arg name ( #4893 )
3 years ago
Ayush Chaurasia
59aae85a7e
W&B: Fix dataset check ( #4879 )
* evolve fix
* Enable login timeout
* fix pkg
* check rank
* don't relogin
* fix
* reformat
3 years ago
Ayush Chaurasia
9febea79de
W&B: Login only in master processes ( #4866 )
* evolve fix
* Enable login timeout
* fix pkg
* check rank
* don't relogin
3 years ago
Glenn Jocher
8ad9e4ed5b
Annotator `check_font()` RANK -1 remove progress ( #4864 )
* Annotator `check_font()` RANK -1 remove progress
* Cleanup
3 years ago
Glenn Jocher
2b3109eeb0
Fix DDP destruction `LOGGER.info()` ( #4863 )
3 years ago
Glenn Jocher
2809616345
Update Dockerfile ( #4862 )
3 years ago
Glenn Jocher
40d1c80503
Update Dockerfile ( #4861 )
3 years ago
Glenn Jocher
9ef94940aa
Update greetings.yml ( #4856 )
3 years ago
Glenn Jocher
4c839eeb10
Simplify `check_requirements()` usage ( #4855 )
* Simplify `check_requirements()` usage
* remove assert, print()
3 years ago
Glenn Jocher
4d1a2ac87e
Update `sys.path.append(str(ROOT))` ( #4852 )
* Update `sys.path.append(str(ROOT))`
* Cleanup
3 years ago
Glenn Jocher
3732f9ac8a
Refactor argparser printing to `print_args()` ( #4850 )
* Refactor argparser printing to `print_args()`
* Cleanup
3 years ago
Glenn Jocher
84bfa89236
Consolidate `init_seeds()` ( #4849 )
3 years ago
Ayush Chaurasia
302a1b0bb0
W&B: Enable login timeout ( #4843 )
* evolve fix
* Enable login timeout
* fix pkg
3 years ago
Glenn Jocher
4fdaec0449
Single `cache_version` definition ( #4846 )
Defines dataset labels *.cache version in a single place, fixing a bug introduced in #4845 .
3 years ago
Glenn Jocher
fe39562a5f
Sorted datasets update to `cache_labels()` ( #4845 )
PR should produce datasets sorted alphabetically by filename. Cache version incremented to 0.5.
Note: will force a one-time re-caching of existing datasets on first-use.
3 years ago
Glenn Jocher
3a822a22ce
`PIL.ImageDraw.text(anchor=...)` removal, reduce to `>=7.1.2` ( #4842 )
* Unpin Pillow
* Update requirements.txt
* Update plots.py
3 years ago
Glenn Jocher
e83792e65c
Created using Colaboratory
3 years ago
Glenn Jocher
850f98f508
Created using Colaboratory
3 years ago
Glenn Jocher
27a4736e96
Fix val.py study plot ( #4831 )
* Fix val.py study plot
* call plot_val_study
* Rename plot_study_txt to plot_val_study
3 years ago
Jiacong Fang
3beb871ba4
Multiple TF export improvements ( #4824 )
* Add fused conv support
* Set all saved_model values to non trainable
* Fix TFLite fp16 model export
* Fix int8 TFLite conversion
3 years ago
Glenn Jocher
6b44ecd53d
Fix 'PyTorch starting from' for URL weights ( #4828 )
Follows #4823
3 years ago
Kalen Michael
43b2817f6e
Feature/fix export on url ( #4823 )
* added callbacks
* added back callback to main
* added save_dir to callback output
* merged in upstream
* removed ghost code
* added url check
* Add url2file()
* Update file-only
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
0dc725e3dc
Refactor `forward()` method profiling ( #4816 )
3 years ago
Glenn Jocher
621b6d5ba8
Evolution `--resume` fix ( #4802 )
Also disable `/weights` dir creation when evolving as no weights are saved and empty folder causes user expectations of weights.
3 years ago
Glenn Jocher
b74dd4ba4f
Add `--int8` argument ( #4799 )
* Add `--int8` argument
* parents[0] bug fix
* Fix order
3 years ago
Jihoon Kim
fcb225c1c8
Shuffle all 4(or 9) images in mosaic augmentation ( #4787 )
Thank you for sharing nice open-source codes 👍
I applied to shuffle the order of all 4(or 9) images in mosaic augmentation
Currently, the order of images in mosaic augmentation is not completely random.
The remaining images except the first are randomly arranged. Apply shuffle all to increase the diversity of data composition.
3 years ago
Glenn Jocher
aa1859909c
Update ci-testing.yml ( #4772 )
3 years ago
Glenn Jocher
b161edf873
Update ci-testing.yml ( #4770 )
3 years ago
Glenn Jocher
c3a93d783d
Add TensorFlow formats to `export.py` ( #4479 )
* Initial commit
* Remove unused export_torchscript return
* ROOT variable
* Add prefix to fcn arg
* fix ROOT
* check_yaml into run()
* interim fixes
* imgsz=(320, 320)
* Hardcode tf_raw_resize False
* Finish opt elimination
* Update representative_dataset_gen()
* Update export.py with TF methods
* SiLU and GraphDef fixes
* file_size() directory handling feature
* export fixes
* add lambda: to representative_dataset
* Detect training False default
* Fuse false for TF models
* Embed agnostic NMS arguments
* Remove lambda
* TensorFlow.js export success
* Add pb to Usage
* Add *_tfjs_model/ to ignore files
* prepend YOLOv5 to function headers
* Remove end --- comments
* parameterize tfjs export pb file
* update run() data default /ROOT
* update --include help
* update imports
* return ct_model
* Consolidate TFLite export
* pb prerequisite to tfjs
* TF modules CamelCase
* Remove exports from tf.py and cleanup
* pass agnostic NMS arguments
* CI
* CI
* ignore *_web_model/
* Add tensorflow to CI dependencies
* CI tensorflow-cpu
* Update requirements.txt
* Remove tensorflow check_requirement
* CI coreml tfjs
* export only onnx torchscript
* reorder exports torchscript first
3 years ago
Glenn Jocher
c47be26f34
Replace `path.absolute()` with `path.resolve()` ( #4763 )
3 years ago
Glenn Jocher
cd810c8286
Centralize `user_config_dir()` decision making ( #4755 )
3 years ago
Glenn Jocher
22ee6fb7c1
Update `is_writeable()` for 2 methods ( #4744 )
* Writeable test
* Fix
* Cleanup
3 years ago
Glenn Jocher
ff35292520
Add TF and TFLite models to `.dockerignore` ( #4748 )
3 years ago
Glenn Jocher
6c554b75ea
Add TF and TFLite models to `.gitignore` ( #4747 )
3 years ago
Glenn Jocher
a144536f88
Fix `is_writeable()` for 3 OS support ( #4743 )
* Fix `is_writeable()` for 3 OS support
* Update general.py
3 years ago
Josh Veitch-Michaelis
19e28e3bfe
Add `callbacks` to train function in W&B sweep ( #4742 )
* add callbacks to train function in wandb sweep
Fix following https://github.com/ultralytics/yolov5/pull/4688 which modified the function signature to `train`
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
7af1b4c266
Improved `detect.py` timing ( #4741 )
* Improved detect.py timing
* Eliminate 1 time_sync() call
* Inference-only time
* dash
* #Save section
* Cleanup
3 years ago
Glenn Jocher
c5360f6e70
Fix `--data from_HUB.zip` ( #4732 )
@KalenMike
3 years ago
Glenn Jocher
4a025ae97f
Fix `user_config_dir()` for GCP/AWS functions ( #4726 )
* Fix `user_config_dir()` for GCP/AWS functions
Compatability fix for GCP functions and AWS lambda for user config directory in https://github.com/ultralytics/yolov5/pull/4628
* Windows skip check
3 years ago
Zegorax
2d9411dbb8
Scope `onnx-simplifier` requirements check ( #4730 )
* Changed onnx-simplifier check behavior
Export.py has been updated to check for onnx-simplifier requirement only when the --simplify argument is added.
Allows for better flexibility and one less requirement if simplify is not needed.
* Fix single-element tuples
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Jean-Baptiste Martin
1cad0ce2c7
Allow `multi_label` option for NMS with PyTorch Hub ( #4728 )
* Allow specifying multi_label option for NMS when using torch hub
* Reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
JD Costa
deb434aefa
Make CONFIG_DIR configurable per environment variable ( #4727 )
3 years ago
ELHoussineT
0d8a184237
Add `crops = results.crops()` dictionary ( #4676 )
* adding get cropped functionality
* Add target logic in existing functions
* Crops cleanup
* Add dictionary keys: conf, cls, box
* Bug fixes - avoid return after first image
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
8e94bf62d9
Add `user_config_dir('Ultralytics')` ( #4715 )
* Add `user_config_dir`
* Linux to .config
3 years ago
Glenn Jocher
25a7e1dae5
Update `check_yaml()` comment ( #4713 )
* Update `check_yaml()` comment
* Cleanup
3 years ago
Glenn Jocher
f984cce52a
Fix `check_suffix()` ( #4712 )
Fix a bug when `file=''`
3 years ago