Commit Graph

52 Commits (4e8c81a368d154fed3f27b16a728b6467ff60c6a)

Author SHA1 Message Date
Glenn Jocher e086347377
Update plots.py () 2021-08-14 12:50:27 +02:00
Glenn Jocher e78aeac973
Evolve in CSV format ()
* Update evolution to CSV format

* Update

* Update

* Update

* Update

* Update

* reset args

* reset args

* reset args

* plot_results() fix

* Cleanup

* Cleanup2
2021-08-04 17:13:38 +02:00
Kalen Michael b74929c910
Add `train.py` and `val.py` callbacks ()
* 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>
2021-08-01 00:18:07 +02:00
Glenn Jocher 96e36a7c91
New CSV Logger ()
* 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
2021-07-25 19:06:37 +02:00
Glenn Jocher efe60b5681
Refactor train.py and val.py `loggers` ()
* 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
2021-07-25 01:18:39 +02:00
Glenn Jocher 720aaa65c8
Rename `test.py` to `val.py` () 2021-07-14 15:43:54 +02:00
Glenn Jocher 248504cf13
Feature visualization improvements 32 () 2021-07-09 15:23:02 +02:00
Glenn Jocher 850970e081
`feature_visualization()` CUDA fix () 2021-07-07 16:23:31 +02:00
Glenn Jocher 87b094bcbc
Feature visualization update ()
* Feature visualization update

* Save to jpg (faster)

* Save to png
2021-07-07 15:41:58 +02:00
Glenn Jocher 02719dde52
Update `feature_visualization()` ()
* Update `feature_visualization()`

Only plot for data with height, width > 1

* cleanup

* Cleanup
2021-06-28 13:48:14 +02:00
Zigarss 20d45aa4f1
Add feature map visualization ()
* 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>
2021-06-28 13:18:45 +02:00
Glenn Jocher 095197bd4a
Ignore Seaborn plot warnings ()
* Ignore Seaborn plot warnings

* Update plots.py

* Update metrics.py
2021-06-10 13:51:29 +02:00
Glenn Jocher 5948f20a3d
Update test.py profiling ()
* Update test.py profiling

* half_precision to half

* inplace
2021-06-09 16:25:17 +02:00
yeric1789 7b0eb952ca
`plot_one_box()` default `color=(128, 128, 128)` ()
* Color can be none by default

* `plot_one_box()` default `color=(128, 128, 128)`

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-19 19:47:36 +02:00
Glenn Jocher ffb47ffbeb
Silent Patches Bug Fix () 2021-05-17 17:47:30 +02:00
Glenn Jocher ae53f50ae7
Plot labels histogram colors () 2021-05-17 11:44:22 +02:00
Glenn Jocher 52c05707d6
Update Colors() ()
* Update Colors()

* update colors

* update colors
2021-05-12 17:41:11 +02:00
Glenn Jocher a833ee2a46
Update check_requirements() exclude list () 2021-04-29 21:16:23 +02:00
Glenn Jocher 57812df68c
New Colors() class () 2021-04-28 16:05:14 +02:00
Glenn Jocher f7bc685c2c
Implement yaml.safe_load() ()
* Implement yaml.safe_load()

* yaml.safe_dump()
2021-04-21 14:34:45 +02:00
Glenn Jocher c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix ()
* 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.
2021-04-18 13:47:40 +02:00
Glenn Jocher f5b8f7d54c
YOLOv5 v5.0 Release () 2021-04-11 19:23:47 +02:00
Glenn Jocher 08d4918d7f
labels.jpg class names ()
* labels.png class names

* fontsize=10
2021-03-12 22:15:41 -08:00
Glenn Jocher c09964c27c
Update inference default to multi_label=False ()
* Update inference default to multi_label=False

* bug fix

* Update plots.py

* Update plots.py
2021-02-19 15:39:09 -08:00
Glenn Jocher 47faf95079 reset head 2021-02-19 15:20:41 -08:00
Glenn Jocher 9646ca438a
Update plot_study() () 2021-02-01 13:51:11 -08:00
Glenn Jocher be9edffded
Update test.py --task speed and study ()
* Add --speed benchmark

* test range 256 - 1536

* update

* update

* update

* update
2021-01-31 13:58:49 -08:00
Glenn Jocher 630ec06af6
Update to colors.TABLEAU_COLORS () 2021-01-28 12:37:21 -08:00
Glenn Jocher 59c21c7bcb
W&B log epoch ()
* 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
2021-01-26 21:16:01 -08:00
Abhiram V 35400dc7b7
Update plots.py with yaml.SafeLoader () 2021-01-18 10:46:20 -08:00
Glenn Jocher dd03b20ba5
colorstr() updates ()
* W&B ImportError message fix

* colorstr() updates

* colorstr() updates

* colorstr() default to 'blue', 'bold'

* train: magenta

* train: blue
2021-01-12 10:33:15 -08:00
Glenn Jocher 69be8e738f
YOLOv5 v4.0 Release ()
* 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
2021-01-04 19:54:09 -08:00
Glenn Jocher 0e341c5660
Create one_cycle() function () 2021-01-04 15:49:08 -08:00
Glenn Jocher 685d601308
Increase plot_labels() speed () 2020-12-18 18:05:38 -08:00
Glenn Jocher 49abc722fc
Update profile_idetection() () 2020-12-18 01:19:17 -08:00
Glenn Jocher 69ea70cd3b
Add idetection_profile() function to plots.py () 2020-12-15 18:35:47 -08:00
Glenn Jocher 84f9bb5d92
Normalized mosaic plotting bug fix () 2020-12-08 18:44:13 -08:00
Glenn Jocher f010147578
Update matplotlib.use('Agg') tight ()
* Update matplotlib tight_layout=True

* udpate

* udpate

* update

* png to ps

* update

* update
2020-12-02 15:53:16 +01:00
Glenn Jocher 784feae30a
Update matplotlib svg backend () 2020-12-02 14:05:12 +01:00
Hu Ye 577f298d9b
plot_images() scale bug fix ()
fix bugs in plot_images
2020-12-01 11:29:59 +01:00
Glenn Jocher b6ed1104a6
Daemon thread plotting ()
* Daemon thread plotting

* remove process_batch

* plot after print
2020-11-30 16:44:14 +01:00
Glenn Jocher 96a84468b9
Update labels_to_image_weights() () 2020-11-28 12:25:45 +01:00
Glenn Jocher c9798ae0e1
Update plot_study_txt() () 2020-11-26 22:18:17 +01:00
Glenn Jocher 0f2057ed33
Targets scaling bug fix () 2020-11-26 18:33:28 +01:00
Glenn Jocher 2c3efa430b
Mosaic plots bug fix () 2020-11-26 14:02:22 +01:00
Glenn Jocher e9a0ae6f19
Cache bug fix ()
* Caching bug fix 

* np.zeros((0,5)) x2
2020-11-25 20:33:14 +01:00
Glenn Jocher 64bce3c822 Update labels.png with rectangles fix () 2020-11-18 12:27:30 +01:00
Glenn Jocher 1542ccadb2
Update labels.png with rectangles () 2020-11-18 00:17:33 +01:00
Glenn Jocher 4250f84dfb
Update PR curve ()
* Update PR curve

* legend outside

* list(Path().glob())
2020-11-17 17:29:55 +01:00
Glenn Jocher 92c9b72832
Matplotlib AGG backend () 2020-11-16 16:34:07 +01:00