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 years ago
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 years ago
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 years ago
Glenn Jocher
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
3 years ago
Glenn Jocher
248504cf13
Feature visualization improvements 32 ( #3947 )
3 years ago
Glenn Jocher
850970e081
`feature_visualization()` CUDA fix ( #3925 )
3 years ago
Glenn Jocher
87b094bcbc
Feature visualization update ( #3920 )
* Feature visualization update
* Save to jpg (faster)
* Save to png
3 years ago
Glenn Jocher
02719dde52
Update `feature_visualization()` ( #3807 )
* Update `feature_visualization()`
Only plot for data with height, width > 1
* cleanup
* Cleanup
3 years ago
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 years ago
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 years ago
Glenn Jocher
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
3 years ago
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 years ago
Glenn Jocher
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
3 years ago
Glenn Jocher
ae53f50ae7
Plot labels histogram colors ( #3192 )
3 years ago
Glenn Jocher
52c05707d6
Update Colors() ( #3046 )
* Update Colors()
* update colors
* update colors
3 years ago
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list ( #2974 )
3 years ago
Glenn Jocher
57812df68c
New Colors() class ( #2963 )
3 years ago
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 years ago
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 years ago
Glenn Jocher
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
3 years ago
Glenn Jocher
08d4918d7f
labels.jpg class names ( #2454 )
* labels.png class names
* fontsize=10
3 years ago
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 years ago
Glenn Jocher
47faf95079
reset head
3 years ago
Glenn Jocher
9646ca438a
Update plot_study() ( #2112 )
3 years ago
Glenn Jocher
be9edffded
Update test.py --task speed and study ( #2099 )
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
3 years ago
Glenn Jocher
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
3 years ago
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 years ago
Abhiram V
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
3 years ago
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 years ago
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 years ago
Glenn Jocher
0e341c5660
Create one_cycle() function ( #1836 )
3 years ago
Glenn Jocher
685d601308
Increase plot_labels() speed ( #1736 )
3 years ago
Glenn Jocher
49abc722fc
Update profile_idetection() ( #1727 )
3 years ago
Glenn Jocher
69ea70cd3b
Add idetection_profile() function to plots.py ( #1700 )
3 years ago
Glenn Jocher
84f9bb5d92
Normalized mosaic plotting bug fix ( #1647 )
3 years ago
Glenn Jocher
f010147578
Update matplotlib.use('Agg') tight ( #1583 )
* Update matplotlib tight_layout=True
* udpate
* udpate
* update
* png to ps
* update
* update
3 years ago
Glenn Jocher
784feae30a
Update matplotlib svg backend ( #1580 )
3 years ago
Hu Ye
577f298d9b
plot_images() scale bug fix ( #1566 )
fix bugs in plot_images
3 years ago
Glenn Jocher
b6ed1104a6
Daemon thread plotting ( #1561 )
* Daemon thread plotting
* remove process_batch
* plot after print
3 years ago
Glenn Jocher
96a84468b9
Update labels_to_image_weights() ( #1545 )
3 years ago
Glenn Jocher
c9798ae0e1
Update plot_study_txt() ( #1533 )
3 years ago
Glenn Jocher
0f2057ed33
Targets scaling bug fix ( #1529 )
3 years ago
Glenn Jocher
2c3efa430b
Mosaic plots bug fix ( #1526 )
3 years ago
Glenn Jocher
e9a0ae6f19
Cache bug fix ( #1513 )
* Caching bug fix #1508
* np.zeros((0,5)) x2
3 years ago
Glenn Jocher
64bce3c822
Update labels.png with rectangles fix ( #1432 )
3 years ago
Glenn Jocher
1542ccadb2
Update labels.png with rectangles ( #1432 )
3 years ago
Glenn Jocher
4250f84dfb
Update PR curve ( #1428 )
* Update PR curve
* legend outside
* list(Path().glob())
3 years ago
Glenn Jocher
92c9b72832
Matplotlib AGG backend ( #1413 )
3 years ago
Glenn Jocher
9c91aeae10
W&B logging add hyperparameters ( #1399 )
* W&B logging add hyperparameters
* hyp bug fix and image logging updates
* if plots and wandb:
* cleanup
* wandb/ gitignore add
* cleanup 2
* cleanup 3
* move wandb import to top of file
* wandb evolve
* update import
* wandb.run.finish()
* default anchors: 3
3 years ago
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
3 years ago