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

1168 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher
cf4f95bc5f
Stale github.actor bug fix (#3483) 2021-06-06 18:06:24 +02:00
Glenn Jocher
3597d280ee
Update stale.yml HUB URL (#3468) 2021-06-04 23:49:43 +02:00
Glenn Jocher
739451da5a
Add additional links/resources to stale.yml message (#3467) 
* Update stale.yml

* cleanup

* Update stale.yml

* reformat
 2021-06-04 23:45:24 +02:00
edificewang
b31229ae89
Revert FP16 test.py and detect.py inference to FP32 default (#3423) 
* fixed inference bug ,while use half precision

* replace --use-half with --half

* replace space and PEP8 in detect.py

* PEP8 detect.py

* update --half help comment

* Update test.py

* revert space

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-04 22:28:34 +02:00
Sam_S
044daafd9d
Add output_names argument for ONNX export with dynamic axes (#3456) 
* Add output names & dynamic axes for onnx export

Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3

* use first output only + cleanup

Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-04 22:01:58 +02:00
Glenn Jocher
317f2ccc9d
Improved check_requirements() offline-handling (#3466) 
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
 2021-06-04 21:34:36 +02:00
Glenn Jocher
563ea9475a
Add check_git_status() 5 second timeout (#3464) 
* Add check_git_status() 5 second timeout

This should prevent the SSH Git bug that we were discussing @KalenMike

* cleanup

* replace timeout with check_output built-in timeout
 2021-06-04 21:17:36 +02:00
Yonghye Kwon
c37f072ba7
Faster HSV augmentation (#3462) 
remove datatype conversion process that can be skipped
 2021-06-04 20:02:20 +02:00
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
4aa2959101
Suppress jit trace warning + graph once (#3454) 
* Suppress jit trace warning + graph once

Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.

* Update train.py
 2021-06-04 12:37:41 +02:00
Glenn Jocher
af2bc3a1c3
Add FLOPs title to table (#3453) 2021-06-04 11:46:40 +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
Glenn Jocher
4b52e19a61
COCO evolution fix (#3388) 
* COCO evolution fix

* cleanup

* update print

* print fix
 2021-05-29 22:49:34 +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
Peretz Cohen
57f773b0ae
Update tutorial.ipynb (#3368) 
add Open in Kaggle badge
 2021-05-29 20:49:24 +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
SkalskiP
b78e30dd2b Merge remote-tracking branch 'origin/develop' into develop 2021-05-28 14:44:45 +02:00
Piotr Skalski
3fea068384 update ci-testing.yml (#3322) 
* update ci-testing.yml

* update greetings.yml

* bring back os matrix
 2021-05-28 14:44:24 +02:00
Piotr Skalski
bb131238aa
update ci-testing.yml (#3322) 
* update ci-testing.yml

* update greetings.yml

* bring back os matrix
 2021-05-27 17:01:36 +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
WangChaofeng
ef4d53818d
ONNX export in .train() mode fix (#3362) 2021-05-27 14:10:14 +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
1f8d716ec9
yolo.py header (#3347) 2021-05-26 12:06:08 +02:00
Glenn Jocher
aad99b63d6
TensorBoard DP/DDP graph fix (#3325) 2021-05-25 11:45:24 +02:00
Glenn Jocher
407dc5008e
Update README.md (#3320) 2021-05-24 17:17:32 +02:00
Glenn Jocher
61ea23c3fe
Implement @torch.no_grad() decorator (#3312) 
* `@torch.no_grad()` decorator

* Update detect.py
 2021-05-24 13:23:09 +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
Glenn Jocher
ee24ae11ea
Comment Cython (#3297) 
Ultralytics REST API fix.
 2021-05-23 17:53:54 +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
dd7f0b7e05
Fix TypeError: 'PosixPath' object is not iterable (#3285) 2021-05-21 23:35:31 +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
Glenn Jocher
10d56d784e
Assert --image-weights not combined with DDP (#3275) 2021-05-21 14:46:42 +02:00
Glenn Jocher
7b36e38cf8
Check CoreML models.train() mode (#3262) 
* Check CoreML models.train() mode

* Update export.py
 2021-05-20 15:22:36 +02:00
Glenn Jocher
7b863a34dc
Add Cython (#3217) 
Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172
 2021-05-19 19:58:53 +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
b7cd1f540d
TensorBoard add_graph() fix (#3236) 2021-05-19 13:51:55 +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