Glenn Jocher
a820b43aca
Automatic Chinese fonts plotting ( #4951 )
* Automatic Chinese fonts plotting
* Default PIL=False
3 年前
Glenn Jocher
5a8e4343d8
Scope `check_file()` search space ( #4933 )
`check_file()` is now limited to searching opt-in directories: /data, /models, /utils. This prevents large non-project directories like /.git and /venv from being searched, which may cause `check_file()` to slow significantly.
3 年前
Glenn Jocher
793383232f
Update default Albumentations ( #4931 )
3 年前
Glenn Jocher
39c17ce0b9
Fix `root` referenced before assignment ( #4920 )
* Fix `root` referenced before assignment
Fix for bug introduced by #4919 discovered on VOC autodownload:
```
python train.py --data VOC.yaml
```
* Cleanup
3 年前
Glenn Jocher
a64a4c839f
Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()` ( #4919 )
* Replace `os.system('unzip file.zip')` -> `ZipFile.extractall()`
* Cleanup
3 年前
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 年前
Ayush Chaurasia
59aae85a7e
W&B: Fix dataset check ( #4879 )
* evolve fix
* Enable login timeout
* fix pkg
* check rank
* don't relogin
* fix
* reformat
3 年前
Ayush Chaurasia
9febea79de
W&B: Login only in master processes ( #4866 )
* evolve fix
* Enable login timeout
* fix pkg
* check rank
* don't relogin
3 年前
Glenn Jocher
8ad9e4ed5b
Annotator `check_font()` RANK -1 remove progress ( #4864 )
* Annotator `check_font()` RANK -1 remove progress
* Cleanup
3 年前
Glenn Jocher
4c839eeb10
Simplify `check_requirements()` usage ( #4855 )
* Simplify `check_requirements()` usage
* remove assert, print()
3 年前
Glenn Jocher
4d1a2ac87e
Update `sys.path.append(str(ROOT))` ( #4852 )
* Update `sys.path.append(str(ROOT))`
* Cleanup
3 年前
Glenn Jocher
3732f9ac8a
Refactor argparser printing to `print_args()` ( #4850 )
* Refactor argparser printing to `print_args()`
* Cleanup
3 年前
Glenn Jocher
84bfa89236
Consolidate `init_seeds()` ( #4849 )
3 年前
Ayush Chaurasia
302a1b0bb0
W&B: Enable login timeout ( #4843 )
* evolve fix
* Enable login timeout
* fix pkg
3 年前
Glenn Jocher
4fdaec0449
Single `cache_version` definition ( #4846 )
Defines dataset labels *.cache version in a single place, fixing a bug introduced in #4845 .
3 年前
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 年前
Glenn Jocher
3a822a22ce
`PIL.ImageDraw.text(anchor=...)` removal, reduce to `>=7.1.2` ( #4842 )
* Unpin Pillow
* Update requirements.txt
* Update plots.py
3 年前
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 年前
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 年前
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 年前
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 年前
Glenn Jocher
c47be26f34
Replace `path.absolute()` with `path.resolve()` ( #4763 )
3 年前
Glenn Jocher
cd810c8286
Centralize `user_config_dir()` decision making ( #4755 )
3 年前
Glenn Jocher
22ee6fb7c1
Update `is_writeable()` for 2 methods ( #4744 )
* Writeable test
* Fix
* Cleanup
3 年前
Glenn Jocher
a144536f88
Fix `is_writeable()` for 3 OS support ( #4743 )
* Fix `is_writeable()` for 3 OS support
* Update general.py
3 年前
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 年前
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 年前
JD Costa
deb434aefa
Make CONFIG_DIR configurable per environment variable ( #4727 )
3 年前
Glenn Jocher
8e94bf62d9
Add `user_config_dir('Ultralytics')` ( #4715 )
* Add `user_config_dir`
* Linux to .config
3 年前
Glenn Jocher
25a7e1dae5
Update `check_yaml()` comment ( #4713 )
* Update `check_yaml()` comment
* Cleanup
3 年前
Glenn Jocher
f984cce52a
Fix `check_suffix()` ( #4712 )
Fix a bug when `file=''`
3 年前
Glenn Jocher
a2b3c71636
Add suffix checks ( #4711 )
* Add suffix checks
* Cleanup
* Cleanup2
* Cleanup3
3 年前
Zhiqiang Wang
8e5f9ddbdb
Remove redundant `ComputeLoss` code ( #4701 )
3 年前
Kalen Michael
2317f86ca4
Optimised Callback Class to Reduce Code and Fix Errors ( #4688 )
* added callbacks
* added back callback to main
* added save_dir to callback output
* reduced code count
* updated callbacks
* added default callback class to main, added missing parameters to on_model_save
* Glenn updates
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 年前
Glenn Jocher
548745181a
EarlyStopper updates ( #4679 )
3 年前
Glenn Jocher
234e8ae6fa
Fix for `plot_evolve()` string argument ( #4639 )
3 年前
Glenn Jocher
de534e9221
Fix 3 for Arial.ttf redownloads with hub inference ( #4629 )
Fix 3 for Arial.ttf redownloads with hub inference, follow-on to #4628 .
3 年前
Glenn Jocher
a4e8f78c5e
Fix 2 for Arial.ttf redownloads with hub inference ( #4628 )
3 年前
Glenn Jocher
ba0f80874f
Fix for Arial.ttf redownloads with hub inference ( #4627 )
3 年前
Glenn Jocher
50a9828679
DDP `torch.jit.trace()` `--sync-bn` fix ( #4615 )
* Remove assert
* debug0
* trace=not opt.sync
* sync to sync_bn fix
* Cleanup
3 年前
Glenn Jocher
bb5ebc290e
Close `matplotlib` plots after opening ( #4612 )
* Close plots
* Replace fig.close() for plt.close()
3 年前
Ayush Chaurasia
11e7c7b48d
W&B: Don't log models in evolve operation ( #4611 )
3 年前
Glenn Jocher
3a72d4a7e3
Update `check_git_status()` warning ( #4610 )
3 年前
Glenn Jocher
e7d1842a98
Auto-download Arial.ttf on init ( #4606 )
* Auto-download Arial.ttf on init
* Fix ROOT
3 年前
Glenn Jocher
dc13820c9d
TensorBoard `on_train_end()` speed improvements ( #4605 )
3 年前
Glenn Jocher
b894e69dfc
Update mosaic plots font size ( #4596 )
3 年前
Glenn Jocher
dbbc6b5c48
Re-order `plots.py` to class-first ( #4595 )
3 年前
Glenn Jocher
e5e5ebc799
Auto-UTF handling ( #4594 )
3 年前
Glenn Jocher
de44376d1b
Create `Annotator()` class ( #4591 )
* Add Annotator() class
* Download Arial
* 2x for loop
* Cleanup
* tuple 2 list
* max_size=1920
* bold logging results to
* tolist()
* im = annotator.im
* PIL save in detect.py
* Smart asarray in detect.py
* revert to cv2.imwrite
* Cleanup
* Return result asarray
* Add `Profile()` profiler
* CamelCase Timeout
* Resize after mosaic
* pillow>=8.0.0
* daemon imwrite
* Add cv2 support
* Remove plot_wh_methods and plot_one_box
* pil=False for hubconf.py annotations
* im.shape bug fix
* colorstr common.py
* join daemons
* Update t.daemon
* Removed daemon saving
3 年前
Takumi Karasawa
7b35971ba5
Fix bug in `plot_one_box` when label is `None` ( #4588 )
3 年前