Glenn Jocher
850970e081
`feature_visualization()` CUDA fix ( #3925 )
2021-07-07 16:23:31 +02:00
Glenn Jocher
87b094bcbc
Feature visualization update ( #3920 )
...
* Feature visualization update
* Save to jpg (faster)
* Save to png
2021-07-07 15:41:58 +02:00
Glenn Jocher
02719dde52
Update `feature_visualization()` ( #3807 )
...
* 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 ( #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>
2021-06-28 13:18:45 +02:00
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
...
* 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 ( #3555 )
...
* 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)` ( #3240 )
...
* 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 ( #3214 )
2021-05-17 17:47:30 +02:00
Glenn Jocher
ae53f50ae7
Plot labels histogram colors ( #3192 )
2021-05-17 11:44:22 +02:00
Glenn Jocher
52c05707d6
Update Colors() ( #3046 )
...
* Update Colors()
* update colors
* update colors
2021-05-12 17:41:11 +02:00
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list ( #2974 )
2021-04-29 21:16:23 +02:00
Glenn Jocher
57812df68c
New Colors() class ( #2963 )
2021-04-28 16:05:14 +02:00
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
...
* 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 ( #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.
2021-04-18 13:47:40 +02:00
Glenn Jocher
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
2021-04-11 19:23:47 +02:00
Glenn Jocher
08d4918d7f
labels.jpg class names ( #2454 )
...
* labels.png class names
* fontsize=10
2021-03-12 22:15:41 -08:00
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
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() ( #2112 )
2021-02-01 13:51:11 -08:00
Glenn Jocher
be9edffded
Update test.py --task speed and study ( #2099 )
...
* 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 ( #2069 )
2021-01-28 12:37:21 -08:00
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
2021-01-26 21:16:01 -08:00
Abhiram V
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
2021-01-18 10:46:20 -08:00
Glenn Jocher
dd03b20ba5
colorstr() updates ( #1909 )
...
* 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 ( #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
2021-01-04 19:54:09 -08:00
Glenn Jocher
0e341c5660
Create one_cycle() function ( #1836 )
2021-01-04 15:49:08 -08:00
Glenn Jocher
685d601308
Increase plot_labels() speed ( #1736 )
2020-12-18 18:05:38 -08:00
Glenn Jocher
49abc722fc
Update profile_idetection() ( #1727 )
2020-12-18 01:19:17 -08:00
Glenn Jocher
69ea70cd3b
Add idetection_profile() function to plots.py ( #1700 )
2020-12-15 18:35:47 -08:00
Glenn Jocher
84f9bb5d92
Normalized mosaic plotting bug fix ( #1647 )
2020-12-08 18:44:13 -08:00
Glenn Jocher
f010147578
Update matplotlib.use('Agg') tight ( #1583 )
...
* 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 ( #1580 )
2020-12-02 14:05:12 +01:00
Hu Ye
577f298d9b
plot_images() scale bug fix ( #1566 )
...
fix bugs in plot_images
2020-12-01 11:29:59 +01:00
Glenn Jocher
b6ed1104a6
Daemon thread plotting ( #1561 )
...
* 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() ( #1545 )
2020-11-28 12:25:45 +01:00
Glenn Jocher
c9798ae0e1
Update plot_study_txt() ( #1533 )
2020-11-26 22:18:17 +01:00
Glenn Jocher
0f2057ed33
Targets scaling bug fix ( #1529 )
2020-11-26 18:33:28 +01:00
Glenn Jocher
2c3efa430b
Mosaic plots bug fix ( #1526 )
2020-11-26 14:02:22 +01:00
Glenn Jocher
e9a0ae6f19
Cache bug fix ( #1513 )
...
* Caching bug fix #1508
* np.zeros((0,5)) x2
2020-11-25 20:33:14 +01:00
Glenn Jocher
64bce3c822
Update labels.png with rectangles fix ( #1432 )
2020-11-18 12:27:30 +01:00
Glenn Jocher
1542ccadb2
Update labels.png with rectangles ( #1432 )
2020-11-18 00:17:33 +01:00
Glenn Jocher
4250f84dfb
Update PR curve ( #1428 )
...
* Update PR curve
* legend outside
* list(Path().glob())
2020-11-17 17:29:55 +01:00
Glenn Jocher
92c9b72832
Matplotlib AGG backend ( #1413 )
2020-11-16 16:34:07 +01:00
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
2020-11-16 13:35:34 +01:00
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
...
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
2020-11-14 11:50:32 +01:00