3a822a22ce
`PIL.ImageDraw.text(anchor=...)` removal, reduce to `>=7.1.2` ( #4842 )
* Unpin Pillow
* Update requirements.txt
* Update plots.py
3 роки тому
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 роки тому
cd810c8286
Centralize `user_config_dir()` decision making ( #4755 )
3 роки тому
deb434aefa
Make CONFIG_DIR configurable per environment variable ( #4727 )
3 роки тому
8e94bf62d9
Add `user_config_dir('Ultralytics')` ( #4715 )
* Add `user_config_dir`
* Linux to .config
3 роки тому
234e8ae6fa
Fix for `plot_evolve()` string argument ( #4639 )
3 роки тому
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 роки тому
a4e8f78c5e
Fix 2 for Arial.ttf redownloads with hub inference ( #4628 )
3 роки тому
bb5ebc290e
Close `matplotlib` plots after opening ( #4612 )
* Close plots
* Replace fig.close() for plt.close()
3 роки тому
e7d1842a98
Auto-download Arial.ttf on init ( #4606 )
* Auto-download Arial.ttf on init
* Fix ROOT
3 роки тому
b894e69dfc
Update mosaic plots font size ( #4596 )
3 роки тому
dbbc6b5c48
Re-order `plots.py` to class-first ( #4595 )
3 роки тому
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 роки тому
7b35971ba5
Fix bug in `plot_one_box` when label is `None` ( #4588 )
3 роки тому
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 роки тому
2da4e7acf7
Merge PIL and OpenCV in `plot_one_box(use_pil=False)` ( #4416 )
* Merge PIL and OpenCV box plotting functions
* Add ASCII check to plot_one_box
* Cleanup
* Cleanup2
3 роки тому
e086347377
Update plots.py ( #4407 )
3 роки тому
e78aeac973
Evolve in CSV format ( #4307 )
* Update evolution to CSV format
* Update
* Update
* Update
* Update
* Update
* reset args
* reset args
* reset args
* plot_results() fix
* Cleanup
* Cleanup2
3 роки тому
b74929c910
Add `train.py` and `val.py` callbacks ( #4220 )
* added callbacks
* Update callbacks.py
* Update train.py
* Update val.py
* Fix CamlCase add staticmethod
* Refactor logger into callbacks
* Cleanup
* New callback on_val_image_end()
* Add curves and results images to TensorBoard
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 роки тому
96e36a7c91
New CSV Logger ( #4148 )
* New CSV Logger
* cleanup
* move batch plots into Logger
* rename comment
* Remove total loss from progress bar
* mloss :-1 bug fix
* Update plot_results()
* Update plot_results()
* plot_results bug fix
3 роки тому
efe60b5681
Refactor train.py and val.py `loggers` ( #4137 )
* Update loggers
* Config
* Update val.py
* cleanup
* fix1
* fix2
* fix3 and reformat
* format sweep.py
* Logger() class
* cleanup
* cleanup2
* wandb package import fix
* wandb package import fix2
* txt fix
* fix4
* fix5
* fix6
* drop wandb into utils/loggers
* fix 7
* rename loggers/wandb_logging to loggers/wandb
* Update message
* Update message
* Update message
* cleanup
* Fix x axis bug
* fix rank 0 issue
* cleanup
3 роки тому
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
3 роки тому
248504cf13
Feature visualization improvements 32 ( #3947 )
3 роки тому
850970e081
`feature_visualization()` CUDA fix ( #3925 )
3 роки тому
87b094bcbc
Feature visualization update ( #3920 )
* Feature visualization update
* Save to jpg (faster)
* Save to png
3 роки тому
02719dde52
Update `feature_visualization()` ( #3807 )
* Update `feature_visualization()`
Only plot for data with height, width > 1
* cleanup
* Cleanup
3 роки тому
20d45aa4f1
Add feature map visualization ( #3804 )
* Add feature map visualization
Add a feature_visualization function to visualize the mid feature map of the model.
* Update yolo.py
* remove boolean from forward and reorder if statement
* remove print from forward
* General cleanup
* Indent
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 роки тому
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 роки тому
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
3 роки тому
7b0eb952ca
`plot_one_box()` default `color=(128, 128, 128)` ( #3240 )
* Color can be none by default
* `plot_one_box()` default `color=(128, 128, 128)`
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 роки тому
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
3 роки тому
ae53f50ae7
Plot labels histogram colors ( #3192 )
3 роки тому
52c05707d6
Update Colors() ( #3046 )
* Update Colors()
* update colors
* update colors
3 роки тому
a833ee2a46
Update check_requirements() exclude list ( #2974 )
3 роки тому
57812df68c
New Colors() class ( #2963 )
3 роки тому
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 роки тому
c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix ( #2831 )
* PyTorch Hub cv2 .save() .show() bug fix
cv2.rectangle() was failing on non-contiguous np array inputs. This checks for contiguous arrays and applies is necessary:
```python
imgs[i] = im if im.data.contiguous else np.ascontiguousarray(im) # update
```
* Update plots.py
```python
assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to plot_on_box() input image.'
```
* Update hubconf.py
Expand CI tests to OpenCV image.
3 роки тому
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
3 роки тому
08d4918d7f
labels.jpg class names ( #2454 )
* labels.png class names
* fontsize=10
3 роки тому
c09964c27c
Update inference default to multi_label=False ( #2252 )
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
3 роки тому
47faf95079
reset head
3 роки тому
9646ca438a
Update plot_study() ( #2112 )
3 роки тому
be9edffded
Update test.py --task speed and study ( #2099 )
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
3 роки тому
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
3 роки тому
59c21c7bcb
W&B log epoch ( #1946 )
* W&B log epoch
* capitalize
* W&B log epoch
* capitalize
* Update train.py
New try using https://docs.wandb.ai/library/log#incremental-logging
* Update train.py
* Update test.py
* Update train.py
* Update plots.py
* Update train.py
* Update train.py
* label plot step -1
* update
* update
* update
* update
* update
* update
* Update train.py
* Update train.py
3 роки тому
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
3 роки тому
dd03b20ba5
colorstr() updates ( #1909 )
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
3 роки тому
69be8e738f
YOLOv5 v4.0 Release ( #1837 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update study.png
* update study.png
* Update datasets.py
3 роки тому
0e341c5660
Create one_cycle() function ( #1836 )
3 роки тому
685d601308
Increase plot_labels() speed ( #1736 )
3 роки тому
Glenn Jocher
JD Costa
Takumi Karasawa
Kalen Michael
Zigarss
yeric1789
Abhiram V