mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,323 Commits 10 Branches 10 Tags
Commit Graph

1323 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher
ee76a68f1d Created using Colaboratory 2021-07-19 13:14:09 +02:00
Glenn Jocher
c8a98cb7cb
Missing nc and names handling in check_dataset() (#4066) 2021-07-19 13:10:21 +02:00
Glenn Jocher
b1be685005
Module super().__init__() (#4065) 
* Module `super().__init__()`

* remove NMS
 2021-07-19 12:41:15 +02:00
Glenn Jocher
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
Glenn Jocher
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
Glenn Jocher
dd62e2d05c
Update README.md (#4041) 
* Update README.md

* Update README.md

* Update README.md
 2021-07-17 23:12:27 +02:00
Glenn Jocher
0067d9578a
Update greetings.yml (#4037) 2021-07-17 15:47:05 +02:00
Glenn Jocher
951922c735
Add --sync-bn known issue (#4032) 
* Add `--sync-bn` known issue

* Update train.py
 2021-07-17 13:07:19 +02:00
Glenn Jocher
b7e985e397
Update greetings.yml (#4024) 
* Update greetings.yml

* Update greetings.yml
 2021-07-16 14:38:59 +02:00
Ayush Chaurasia
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
Glenn Jocher
720aaa65c8
Rename test.py to val.py (#4000) 2021-07-14 15:43:54 +02:00
Glenn Jocher
8ee9fd1505
Update README.md (#3996) 2021-07-13 23:07:09 +02:00
Glenn Jocher
d204a61834
Alert (no detections) (#3984) 
* `Detections()` class `print()` overload

* Update common.py
 2021-07-12 19:48:47 +02:00
Glenn Jocher
b3dabdcc38
Update probability to p (#3980) 2021-07-12 15:54:43 +02:00
Glenn Jocher
41fdf9fa53 Created using Colaboratory 2021-07-12 12:43:26 +02:00
KEN
647223a7a8
Ensemble() visualize fix (#3973) 
* fix visualize error

* Revert "fix visualize error"

* add visualise profile
 2021-07-11 19:47:08 +02:00
Glenn Jocher
a544d59f52
Update requirements.txt albumentations>=1.0.2 (#3972) 2021-07-11 19:07:42 +02:00
Glenn Jocher
90e60b403d
Update np.random.random() to random.random() (#3967) 2021-07-11 16:56:36 +02:00
Glenn Jocher
8298ce5e88
Update albumentations>=1.0.2 (#3966) 2021-07-11 15:49:30 +02:00
Glenn Jocher
80299a57e2
Numerical stability fix for Albumentations (#3958) 2021-07-10 19:50:53 +02:00
Glenn Jocher
443af8b25a
Cache v0.4 update (#3954) 2021-07-10 14:18:46 +02:00
Glenn Jocher
a26e7de2bf
Update augmentations.py (#3948) 2021-07-09 16:45:04 +02:00
Glenn Jocher
248504cf13
Feature visualization improvements 32 (#3947) 2021-07-09 15:23:02 +02:00
Glenn Jocher
dabad5793a Created using Colaboratory 2021-07-08 15:45:53 +02:00
Eldar Kurtic
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
Glenn Jocher
8c6f9e15bf
Update dataset_stats() for zipped datasets (#3926) 
* Update `dataset_stats()` for zipped datasets

@KalenMike

* cleanup
 2021-07-08 11:42:30 +02:00
Glenn Jocher
850970e081
feature_visualization() CUDA fix (#3925) 2021-07-07 16:23:31 +02:00
jmiranda-laplateforme
588094eb7a
Update setattr() default for Hub PIL images (#3923) 
Fix inference from PIL source.
 2021-07-07 16:13:12 +02:00
Glenn Jocher
411842e058
Fix torch.hub.list('ultralytics/yolov5') pathlib bug (#3921) 2021-07-07 16:08:42 +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
johnohagan
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
Glenn Jocher
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
Glenn Jocher
6a3ee7cf03
Hub models map_location=device (#3894) 
* Hub models `map_location=device`

* cleanup
 2021-07-05 16:20:46 +02:00
Glenn Jocher
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
Glenn Jocher
3c3f8fbd5d
Improved BGR2RGB speeds (#3880) 
* Update BGR2RGB ops

* speed improvements

* cleanup
 2021-07-04 20:12:32 +02:00
Glenn Jocher
9e8fb9fd0b
Create utils/augmentations.py (#3877) 
* Create `utils/augmentations.py`

* cleanup
 2021-07-04 18:14:04 +02:00
Glenn Jocher
81b31824f5
Models *.yaml reformat (#3875) 2021-07-04 16:55:08 +02:00
Glenn Jocher
bd88e7f4f2 Created using Colaboratory 2021-07-04 15:55:23 +02:00
ketan-b
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
san-soucie
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
Valentin Aliferov
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
Glenn Jocher
4717a3b038 Created using Colaboratory 2021-07-02 12:24:27 +02:00
Glenn Jocher
b6863385b5 Created using Colaboratory 2021-07-02 12:23:09 +02:00
Glenn Jocher
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
Glenn Jocher
25d1f2932c
Plot AutoShape() detections in ascending order (#3843) 2021-06-30 15:10:40 +02:00
Feras Oughali
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
57c5d02bbe
Concise TransformerBlock() (#3821) 2021-06-29 16:03:10 +02:00
Glenn Jocher
5ea771d93d
Move IoU functions to metrics.py (#3820) 2021-06-29 13:18:13 +02:00
Glenn Jocher
3213d8713f
Fix for dataset_stats() with updated data.yaml (#3819) 
@KalenMike
 2021-06-29 12:44:59 +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