Glenn Jocher
e7d1842a98
Auto-download Arial.ttf on init ( #4606 )
* Auto-download Arial.ttf on init
* Fix ROOT
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
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 роки тому
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 роки тому
Glenn Jocher
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 роки тому
Glenn Jocher
e086347377
Update plots.py ( #4407 )
3 роки тому
Glenn Jocher
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 роки тому
Kalen Michael
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 роки тому
Glenn Jocher
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 роки тому
Glenn Jocher
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 роки тому
Glenn Jocher
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
3 роки тому
Glenn Jocher
248504cf13
Feature visualization improvements 32 ( #3947 )
3 роки тому
Glenn Jocher
850970e081
`feature_visualization()` CUDA fix ( #3925 )
3 роки тому
Glenn Jocher
87b094bcbc
Feature visualization update ( #3920 )
* Feature visualization update
* Save to jpg (faster)
* Save to png
3 роки тому
Glenn Jocher
02719dde52
Update `feature_visualization()` ( #3807 )
* Update `feature_visualization()`
Only plot for data with height, width > 1
* cleanup
* Cleanup
3 роки тому
Zigarss
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 роки тому
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 роки тому
Glenn Jocher
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
3 роки тому
yeric1789
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 роки тому
Glenn Jocher
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
3 роки тому
Glenn Jocher
ae53f50ae7
Plot labels histogram colors ( #3192 )
3 роки тому
Glenn Jocher
52c05707d6
Update Colors() ( #3046 )
* Update Colors()
* update colors
* update colors
3 роки тому
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list ( #2974 )
3 роки тому
Glenn Jocher
57812df68c
New Colors() class ( #2963 )
3 роки тому
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 роки тому
Glenn Jocher
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 роки тому
Glenn Jocher
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
3 роки тому
Glenn Jocher
08d4918d7f
labels.jpg class names ( #2454 )
* labels.png class names
* fontsize=10
3 роки тому
Glenn Jocher
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 роки тому
Glenn Jocher
47faf95079
reset head
3 роки тому
Glenn Jocher
9646ca438a
Update plot_study() ( #2112 )
3 роки тому
Glenn Jocher
be9edffded
Update test.py --task speed and study ( #2099 )
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
3 роки тому
Glenn Jocher
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
3 роки тому
Glenn Jocher
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 роки тому
Abhiram V
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
3 роки тому
Glenn Jocher
dd03b20ba5
colorstr() updates ( #1909 )
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
3 роки тому
Glenn Jocher
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 роки тому
Glenn Jocher
0e341c5660
Create one_cycle() function ( #1836 )
3 роки тому
Glenn Jocher
685d601308
Increase plot_labels() speed ( #1736 )
3 роки тому
Glenn Jocher
49abc722fc
Update profile_idetection() ( #1727 )
3 роки тому
Glenn Jocher
69ea70cd3b
Add idetection_profile() function to plots.py ( #1700 )
3 роки тому
Glenn Jocher
84f9bb5d92
Normalized mosaic plotting bug fix ( #1647 )
3 роки тому
Glenn Jocher
f010147578
Update matplotlib.use('Agg') tight ( #1583 )
* Update matplotlib tight_layout=True
* udpate
* udpate
* update
* png to ps
* update
* update
3 роки тому
Glenn Jocher
784feae30a
Update matplotlib svg backend ( #1580 )
3 роки тому
Hu Ye
577f298d9b
plot_images() scale bug fix ( #1566 )
fix bugs in plot_images
3 роки тому
Glenn Jocher
b6ed1104a6
Daemon thread plotting ( #1561 )
* Daemon thread plotting
* remove process_batch
* plot after print
3 роки тому
Glenn Jocher
96a84468b9
Update labels_to_image_weights() ( #1545 )
3 роки тому
Glenn Jocher
c9798ae0e1
Update plot_study_txt() ( #1533 )
3 роки тому