3bef77f5cb
Addition refactor `export.py` ( #4089 )
...
* Addition refactor `export.py`
* Update export.py
2021-07-20 18:42:27 +02:00
442a7abdf2
Refactor `export.py` ( #4080 )
...
* Refactor `export.py`
* cleanup
* Update check_requirements()
* Update export.py
2021-07-20 13:21:52 +02:00
0cc7c58787
W&B: fix refactor bugs ( #4069 )
2021-07-19 13:57:13 +02:00
7fdcc77bf4
Albumentations >= 1.0.3 ( #4068 )
2021-07-19 13:23:19 +02:00
ee76a68f1d
Created using Colaboratory
2021-07-19 13:14:09 +02:00
c8a98cb7cb
Missing `nc` and `names` handling in check_dataset() ( #4066 )
2021-07-19 13:10:21 +02:00
b1be685005
Module `super().__init__()` ( #4065 )
...
* Module `super().__init__()`
* remove NMS
2021-07-19 12:41:15 +02:00
f7d8562060
`val.py` refactor ( #4053 )
...
* val.py refactor
* cleanup
* cleanup
* cleanup
* cleanup
* save after eval
* opt.imgsz bug fix
* wandb refactor
* dataloader to train_loader
* capitalize global variables
* runs/hub/exp to runs/detect/exp
* refactor wandb logging
* Refactor wandb operations (#4061 )
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2021-07-19 10:43:01 +02:00
9dd33fd20f
AutoShape PosixPath support ( #4047 )
...
* AutoShape PosixPath support
Usage example:
````python
from pathlib import Path
model = ...
file = Path('data/images/zidane.jpg')
results = model(file)
```
* Update common.py
2021-07-18 15:25:37 +02:00
dd62e2d05c
Update README.md ( #4041 )
...
* Update README.md
* Update README.md
* Update README.md
2021-07-17 23:12:27 +02:00
0067d9578a
Update greetings.yml ( #4037 )
2021-07-17 15:47:05 +02:00
951922c735
Add `--sync-bn` known issue ( #4032 )
...
* Add `--sync-bn` known issue
* Update train.py
2021-07-17 13:07:19 +02:00
b7e985e397
Update greetings.yml ( #4024 )
...
* Update greetings.yml
* Update greetings.yml
2021-07-16 14:38:59 +02:00
62409eea08
W&B sweeps support ( #3938 )
...
* Add support for W&B Sweeps
* Update and reformat
* Update search space
* reformat
* reformat sweep.py
* Update sweep.py
* Move sweeps files to wandb dir
* Remove print
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-14 19:13:02 +02:00
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
2021-07-14 15:43:54 +02:00
8ee9fd1505
Update README.md ( #3996 )
2021-07-13 23:07:09 +02:00
d204a61834
Alert (no detections) ( #3984 )
...
* `Detections()` class `print()` overload
* Update common.py
2021-07-12 19:48:47 +02:00
b3dabdcc38
Update `probability` to `p` ( #3980 )
2021-07-12 15:54:43 +02:00
41fdf9fa53
Created using Colaboratory
2021-07-12 12:43:26 +02:00
647223a7a8
`Ensemble()` visualize fix ( #3973 )
...
* fix visualize error
* Revert "fix visualize error"
* add visualise profile
2021-07-11 19:47:08 +02:00
a544d59f52
Update requirements.txt `albumentations>=1.0.2` ( #3972 )
2021-07-11 19:07:42 +02:00
90e60b403d
Update `np.random.random()` to `random.random()` ( #3967 )
2021-07-11 16:56:36 +02:00
8298ce5e88
Update `albumentations>=1.0.2` ( #3966 )
2021-07-11 15:49:30 +02:00
80299a57e2
Numerical stability fix for Albumentations ( #3958 )
2021-07-10 19:50:53 +02:00
443af8b25a
Cache v0.4 update ( #3954 )
2021-07-10 14:18:46 +02:00
a26e7de2bf
Update augmentations.py ( #3948 )
2021-07-09 16:45:04 +02:00
248504cf13
Feature visualization improvements 32 ( #3947 )
2021-07-09 15:23:02 +02:00
dabad5793a
Created using Colaboratory
2021-07-08 15:45:53 +02:00
e7888af94c
Fix inconsistent NMS IoU value for COCO ( #3934 )
...
Evaluation of 'best' and 'last' models will use the same params as the evaluation during the training phase.
This PR fixes https://github.com/ultralytics/yolov5/issues/3907
2021-07-08 15:29:02 +02:00
8c6f9e15bf
Update `dataset_stats()` for zipped datasets ( #3926 )
...
* Update `dataset_stats()` for zipped datasets
@KalenMike
* cleanup
2021-07-08 11:42:30 +02:00
850970e081
`feature_visualization()` CUDA fix ( #3925 )
2021-07-07 16:23:31 +02:00
588094eb7a
Update `setattr()` default for Hub PIL images ( #3923 )
...
Fix inference from PIL source.
2021-07-07 16:13:12 +02:00
411842e058
Fix `torch.hub.list('ultralytics/yolov5')` pathlib bug ( #3921 )
2021-07-07 16:08:42 +02:00
87b094bcbc
Feature visualization update ( #3920 )
...
* Feature visualization update
* Save to jpg (faster)
* Save to png
2021-07-07 15:41:58 +02:00
61047a2b4f
Save PyTorch Hub models to `/root/hub/cache/dir` ( #3904 )
...
* Create hubconf.py
* Add save_dir variable
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-07 13:41:46 +02:00
33202b7f0b
YOLOv5 + Albumentations integration ( #3882 )
...
* Albumentations integration
* ToGray p=0.01
* print confirmation
* create instance in dataloader init method
* improved version handling
* transform not defined fix
* assert string update
* create check_version()
* add spaces
* update class comment
2021-07-05 18:01:54 +02:00
6a3ee7cf03
Hub models `map_location=device` ( #3894 )
...
* Hub models `map_location=device`
* cleanup
2021-07-05 16:20:46 +02:00
8930e22cce
Evolution commented `hyp['anchors']` fix ( #3887 )
...
Fix for `KeyError: 'anchors'` error when start hyperparameter evolution:
```bash
python train.py --evolve
```
```bash
Traceback (most recent call last):
File "E:\yolov5\train.py", line 623, in <module>
hyp[k] = max(hyp[k], v[1]) # lower limit
KeyError: 'anchors'
```
2021-07-05 12:48:27 +02:00
3c3f8fbd5d
Improved BGR2RGB speeds ( #3880 )
...
* Update BGR2RGB ops
* speed improvements
* cleanup
2021-07-04 20:12:32 +02:00
9e8fb9fd0b
Create `utils/augmentations.py` ( #3877 )
...
* Create `utils/augmentations.py`
* cleanup
2021-07-04 18:14:04 +02:00
81b31824f5
Models `*.yaml` reformat ( #3875 )
2021-07-04 16:55:08 +02:00
bd88e7f4f2
Created using Colaboratory
2021-07-04 15:55:23 +02:00
9d86b54eb3
Add multi-stream saving feature ( #3864 )
...
* Added the recording feature for multiple streams
Thanks for the very cool repo!!
I was trying to record multiple feeds at the same time, but the current version of the detector only had one video writer and one vid_path!
So the streams were not being saved and only were initialized with one frame and this process didn't record the whole thing.
Fix:
I made a list of `vid_writer` and `vid_path` and the `i` from the loop over the `pred` took care of the writer which need to work!
I hope this helps, Thanks!
* Cleanup list lengths
* batch size variable
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-04 12:55:57 +02:00
d3e9d69850
`--evolve 300` generations CLI argument ( #3863 )
...
* evolve command accepts argument for number of generations
* evolve generations argument used in evolve for loop
* evolve argument boolean fixes
* default to 300 evolve generations
* Update train.py
Co-authored-by: John San Soucie <jsansoucie@whoi.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-04 12:14:35 +02:00
831773f5a2
Add EXIF rotation to YOLOv5 Hub inference ( #3852 )
...
* rotating an image according to its exif tag
* Update common.py
* Update datasets.py
* Update datasets.py
faster
* delete extraneous gpg file
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-02 13:25:54 +02:00
4717a3b038
Created using Colaboratory
2021-07-02 12:24:27 +02:00
b6863385b5
Created using Colaboratory
2021-07-02 12:23:09 +02:00
c6c88dc601
Copy-Paste augmentation for YOLOv5 ( #3845 )
...
* Copy-paste augmentation initial commit
* if any segments
* Add obscuration rejection
* Add copy_paste hyperparameter
* Update comments
2021-07-01 00:35:04 +02:00
25d1f2932c
Plot `AutoShape()` detections in ascending order ( #3843 )
2021-06-30 15:10:40 +02:00
7d6af69638
Fix `LoadStreams()` dataloader frame skip issue ( #3833 )
...
* Update datasets.py to read every 4th frame of streams
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-06-30 12:11:29 +02:00
Glenn Jocher
Ayush Chaurasia
KEN
Eldar Kurtic
jmiranda-laplateforme
johnohagan
ketan-b
san-soucie
Valentin Aliferov
Feras Oughali