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,160 Commits 10 Branches 10 Tags
Commit Graph

424 Commits

Author SHA1 Message Date
Glenn Jocher
d40481acc5
Add timeout() class (#3460) 
* Add `timeout()` class

* rearrange order
 2021-06-04 15:23:33 +02:00
Glenn Jocher
8e3b4a0bf3
Update MixUp augmentation alpha=beta=32.0 (#3455) 
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
 2021-06-04 12:47:53 +02:00
Glenn Jocher
f8651c388f
Parse URL authentication (#3424) 
* Parse URL authentication

* urllib.parse.unquote()

* improved error handling

* improved error handling

* remove %3F

* update check_file()
 2021-06-03 18:44:58 +02:00
chocosaj
3cb9ad4fc4
Update FLOPs description (#3422) 
* Update README.md

* Changing FLOPS to FLOPs.

Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
 2021-06-03 12:31:51 +02:00
Glenn Jocher
fdbe527dc0
Revert "cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379)" (#3395) 
This reverts commit 21a9607e00f1365b21d8c4bd81bdbf5fc0efea24.
 2021-05-31 10:39:00 +02:00
Glenn Jocher
d833ab3d25
Create is_pip() function (#3391) 
Returns `True` if file is part of pip package. Useful for contextual behavior modification.

```python
def is_pip():
    # Is file in a pip package?
    return 'site-packages' in Path(__file__).absolute().parts
```
 2021-05-30 20:52:42 +02:00
tudoulei
21a9607e00
cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379) 
* Update datasets.py

* comment

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-29 21:12:01 +02:00
Glenn Jocher
ba6f3f974b
Enable direct --weights URL definition (#3373) 
* Enable direct `--weights URL` definition

@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```

* cleanup

* bug fixes

* weights = attempt_download(weights)

* Update experimental.py

* Update hubconf.py

* return bug fix

* comment mirror

* min_bytes
 2021-05-28 15:18:44 +02:00
Glenn Jocher
4d4a2b0520
Ignore blank lines in *.txt labels (#3366) 
Fix for https://github.com/ultralytics/yolov5/issues/958#issuecomment-849512083
 2021-05-27 14:31:26 +02:00
Glenn Jocher
2435bfe896
Add URL download to check_file() (#3330) 
* Add URL file download to check_file()

* cleanup

* pathlib bug fix
 2021-05-26 15:51:49 +02:00
Glenn Jocher
c6b5bfca85
Updated cache v0.2 with hashlib (#3350) 
* Update cache v0.2 to include parent hash

Possible fix for https://github.com/ultralytics/yolov5/issues/3349

* Update datasets.py
 2021-05-26 14:26:52 +02:00
Glenn Jocher
aad99b63d6
TensorBoard DP/DDP graph fix (#3325) 2021-05-25 11:45:24 +02:00
Glenn Jocher
73a92dc1b6
Explicit git clone master (#3311) 2021-05-24 12:42:36 +02:00
Glenn Jocher
9f3a388cea
Improved check_requirements() robustness (#3298) 
Add try: except clause on missing requirements install to catch install failures. Prompted by Ultralytics YOLOv5 API failure on Cython.
 2021-05-23 18:00:23 +02:00
yeric1789
a039b7796d
PyTorch Hub crops = results.crop() return values (#3282) 
* Changing save_one_box 

Made to work with other changes to common.py

* PEP8 and single line BGR

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-23 16:00:55 +02:00
Glenn Jocher
0e2f2cbb51
Update LoadStreams init fallbacks (#3295) 2021-05-23 14:55:42 +02:00
Charles Frye
19100ba007
Improves docs and handling of entities and resuming by WandbLogger (#3264) 
* adds latest tag to match wandb defaults

* adds entity handling, 'last' tag

* fixes bug causing finished runs to resume

* removes redundant "last" tag for wandb artifact
 2021-05-21 23:42:53 +02:00
Glenn Jocher
683cefead4
YouTube stream ending fix (#3277) 
* YouTube stream ending fix

Properly terminates YouTube streams on video end. Should resolve issues #2769 and #3220.

* Update datasets.py
 2021-05-21 16:51:07 +02:00
Glenn Jocher
f3402353fb
check batch_size % utilized_device_count (#3276) 
Bug fix to check batch_size divisibility of utilized CUDA device count vs total system CUDA device count.
 2021-05-21 15:20:03 +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
Adrian Holovaty
3f74cd9ed1
Parameterize max_det + inference default at 1000 (#3215) 
* Added max_det parameters in various places

* 120 character line

* PEP8

* 120 character line

* Update inference default to 1000 instances

* Update inference default to 1000 instances

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-17 22:47:44 +02:00
Glenn Jocher
13a1c72699
Update datasets.py (#3216) 2021-05-17 22:24:26 +02:00
Glenn Jocher
ffb47ffbeb
Silent Patches Bug Fix (#3214) 2021-05-17 17:47:30 +02:00
Glenn Jocher
36b0a9e7d0
Add CAP_PROP_FRAME_COUNT for YouTube sources (#3193) 2021-05-17 12:27:40 +02:00
Glenn Jocher
ae53f50ae7
Plot labels histogram colors (#3192) 2021-05-17 11:44:22 +02:00
Glenn Jocher
c1c7eb023f
Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
Glenn Jocher
52c05707d6
Update Colors() (#3046) 
* Update Colors()

* update colors

* update colors
 2021-05-12 17:41:11 +02:00
Glenn Jocher
abfcf9eb79
Replace print() with logging.info() in trainloader (#3103) 
Might indirectly help #3095 by providing better visibility on source of corruption.
 2021-05-10 17:07:16 +02:00
Glenn Jocher
57b0d3a605
Add check_python() (#3088) 
* Add check_python()

Checks python version against minimum version of 3.7.0.

* remove packaging dependency

* refactor import
 2021-05-09 15:33:38 +02:00
Ayush Chaurasia
d2a17289c9
Explicitly convert artifact path to posix_path (#3067) 
* Explicitly convert artifact path to posix_path

* Remove redudant str() casting
 2021-05-07 19:53:56 +02:00
Ayush Chaurasia
b18ca31ff5
use check_file for --data (#3035) 2021-05-05 14:28:19 +02:00
Glenn Jocher
8cab907f60
download() ThreadPool update (#3027) 
* download() ThreadPool update

* update train image count

* cid + 1
 2021-05-04 23:37:31 +02:00
Glenn Jocher
5189b3addb Objects365 Update 2021-05-04 02:25:14 +02:00
Glenn Jocher
41cc7caee6
Add NMS threshold checks (#3020) 
* Add NMS threshold checks

* fix
 2021-05-03 15:07:20 +02:00
Glenn Jocher
fcd57028d2
Add is_colab() function (#3018) 2021-05-03 12:37:50 +02:00
Glenn Jocher
31ee54c28c
Hubconf.py bug fix (#3007) 2021-05-01 18:15:17 +02:00
Glenn Jocher
37eaffec7d
Curl update (#3004) 
* Curl update

* Curl update
 2021-05-01 17:29:51 +02:00
Glenn Jocher
4d7bca7648
Add unzip flag to download() (#3002) 2021-05-01 12:37:41 +02:00
Glenn Jocher
54652fe3ff Objects365 update 2021-05-01 00:16:58 +02:00
Ayush Chaurasia
801b469878
Improve performance of dataset Logger (#2943) 
* Improve performance of Dataset Logger

* Fix scaling bug
 2021-04-30 21:54:26 +02:00
Glenn Jocher
955eea8b96
Make cache saving optional (#2977) 2021-04-30 00:56:44 +02:00
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
Glenn Jocher
2c7c075fb1
Update restapi.py (#2967) 2021-04-28 17:50:07 +02:00
Glenn Jocher
57812df68c
New Colors() class (#2963) 2021-04-28 16:05:14 +02:00
Glenn Jocher
1849916726
Check_requirements() enclosing apostrophe bug fix (#2929) 
This fixes a bug where the '>' symbol in python package requirements was not running correctly with subprocess.check_output() commands.
 2021-04-25 23:33:25 +02:00
Glenn Jocher
9c7bb5a52c
ACON Activation batch-size 1 bug patch (#2901) 
* ACON Activation batch-size 1 bug path

This is not a great solution to https://github.com/nmaac/acon/issues/4 but it's all I could think of at the moment.

WARNING: YOLOv5 models with MetaAconC() activations are incapable of running inference at batch-size 1 properly due to a known bug in https://github.com/nmaac/acon/issues/4 with no known solution.

* Update activations.py

* Update activations.py

* Update activations.py

* Update activations.py
 2021-04-25 22:54:07 +02:00
NanoCode012
c0d3f80544
Add verbose option to pytorch hub models (#2926) 
* Add verbose and update print to logging

* Fix positonal param

* Revert auto formatting changes

* Update hubconf.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-04-25 22:01:05 +02:00
Glenn Jocher
45632b2704
Update download() for tar.gz files (#2919) 
* Update download() for tar.gz files

* Update general.py
 2021-04-24 15:53:15 +02:00
Glenn Jocher
1b1ab4cca2
Add file_size() function (#2911) 
* Add file_size() function

* Update export.py
 2021-04-24 01:31:11 +02:00
Glenn Jocher
14d2d2d75f
Update google_utils.py (#2900) 2021-04-22 20:27:32 +02:00