zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
1,383 Commits 5 Branches 0 Tags 33 MiB
Commit Graph

1383 Commits

Author SHA1 Message Date
Glenn Jocher 264be1a616
Rename `opset_version` to `opset` (#4135) 2021-07-24 13:19:12 +02:00
Glenn Jocher 2e538443b7
ONNX inference update (#4073) 2021-07-24 13:08:51 +02:00
Glenn Jocher 39ef6c7a80
Update README.md (#4134) 2021-07-24 12:36:07 +02:00
imyhxy 4495e00016
Fix indentation in `log_training_progress()` (#4126) 2021-07-23 14:55:00 +02:00
Glenn Jocher 4bad914761
Update resume.py (#4115) 2021-07-22 17:22:11 +02:00
Glenn Jocher 2c073cd207
Add train.py ``--img-size` floor (#4099) 2021-07-21 16:50:47 +02:00
Glenn Jocher 3bef77f5cb
Addition refactor `export.py` (#4089) 
* Addition refactor `export.py`

* Update export.py
 2021-07-20 18:42:27 +02:00
Glenn Jocher 442a7abdf2
Refactor `export.py` (#4080) 
* Refactor `export.py`

* cleanup

* Update check_requirements()

* Update export.py
 2021-07-20 13:21:52 +02:00
Ayush Chaurasia 0cc7c58787
W&B: fix refactor bugs (#4069) 2021-07-19 13:57:13 +02:00
Glenn Jocher 7fdcc77bf4
Albumentations >= 1.0.3 (#4068) 2021-07-19 13:23:19 +02:00
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