Yonghye Kwon
8d1ddc93c7
Earlier `assert` for cpu and half option ( #3508 )
* early assert for cpu and half option
early assert for cpu and half option
* Modified comment
Modified comment
3 years ago
Glenn Jocher
3f03acb3db
check_requirements() exclude `opencv-python` ( #3507 )
Duplicate of #3495 merged to `develop`. This PR will be merged to `master`.
Fixes https://github.com/ultralytics/yolov5/issues/3494 .
3 years ago
Glenn Jocher
90b7895d65
check_requirements() exclude `opencv-python` ( #3495 )
Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494 .
3 years ago
Yonghye Kwon
a1c3572bc9
Explicit `model.eval()` call `if opt.train=False` ( #3475 )
* call model.eval() when opt.train is False
call model.eval() when opt.train is False
* single-line if statement
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
cf4f95bc5f
Stale `github.actor` bug fix ( #3483 )
3 years ago
Glenn Jocher
3597d280ee
Update stale.yml HUB URL ( #3468 )
3 years ago
Glenn Jocher
739451da5a
Add additional links/resources to stale.yml message ( #3467 )
* Update stale.yml
* cleanup
* Update stale.yml
* reformat
3 years ago
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>
3 years ago
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>
3 years ago
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).
3 years ago
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
3 years ago
Yonghye Kwon
c37f072ba7
Faster HSV augmentation ( #3462 )
remove datatype conversion process that can be skipped
3 years ago
Glenn Jocher
d40481acc5
Add `timeout()` class ( #3460 )
* Add `timeout()` class
* rearrange order
3 years ago
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
3 years ago
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
3 years ago
Glenn Jocher
af2bc3a1c3
Add FLOPs title to table ( #3453 )
3 years ago
Glenn Jocher
f8651c388f
Parse URL authentication ( #3424 )
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
3 years ago
chocosaj
3cb9ad4fc4
Update FLOPs description ( #3422 )
* Update README.md
* Changing FLOPS to FLOPs.
Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
3 years ago
Glenn Jocher
fdbe527dc0
Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED ( #3379 )" ( #3395 )
This reverts commit 21a9607e00
.
3 years ago
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
```
3 years ago
Glenn Jocher
4b52e19a61
COCO evolution fix ( #3388 )
* COCO evolution fix
* cleanup
* update print
* print fix
3 years ago
tudoulei
21a9607e00
`cv2.imread(img, -1)` for IMREAD_UNCHANGED ( #3379 )
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Peretz Cohen
57f773b0ae
Update tutorial.ipynb ( #3368 )
add Open in Kaggle badge
3 years ago
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
3 years ago
SkalskiP
b78e30dd2b
Merge remote-tracking branch 'origin/develop' into develop
3 years ago
Piotr Skalski
3fea068384
update ci-testing.yml ( #3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
3 years ago
Piotr Skalski
bb131238aa
update ci-testing.yml ( #3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
3 years ago
Glenn Jocher
4d4a2b0520
Ignore blank lines in `*.txt` labels ( #3366 )
Fix for https://github.com/ultralytics/yolov5/issues/958#issuecomment-849512083
3 years ago
WangChaofeng
ef4d53818d
ONNX export in .train() mode fix ( #3362 )
3 years ago
Glenn Jocher
2435bfe896
Add URL download to check_file() ( #3330 )
* Add URL file download to check_file()
* cleanup
* pathlib bug fix
3 years ago
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
3 years ago
Glenn Jocher
1f8d716ec9
yolo.py header ( #3347 )
3 years ago
Glenn Jocher
aad99b63d6
TensorBoard DP/DDP graph fix ( #3325 )
3 years ago
Glenn Jocher
407dc5008e
Update README.md ( #3320 )
3 years ago
Glenn Jocher
61ea23c3fe
Implement `@torch.no_grad()` decorator ( #3312 )
* `@torch.no_grad()` decorator
* Update detect.py
3 years ago
Glenn Jocher
73a92dc1b6
Explicit `git clone` master ( #3311 )
3 years ago
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.
3 years ago
Glenn Jocher
ee24ae11ea
Comment Cython ( #3297 )
Ultralytics REST API fix.
3 years ago
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>
3 years ago
Glenn Jocher
0e2f2cbb51
Update LoadStreams init fallbacks ( #3295 )
3 years ago
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
3 years ago
Glenn Jocher
dd7f0b7e05
Fix TypeError: 'PosixPath' object is not iterable ( #3285 )
3 years ago
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
3 years ago
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.
3 years ago
Glenn Jocher
10d56d784e
Assert `--image-weights` not combined with DDP ( #3275 )
3 years ago
Glenn Jocher
7b36e38cf8
Check CoreML models.train() mode ( #3262 )
* Check CoreML models.train() mode
* Update export.py
3 years ago
Glenn Jocher
7b863a34dc
Add Cython ( #3217 )
Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172
3 years ago
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>
3 years ago
Glenn Jocher
b7cd1f540d
TensorBoard add_graph() fix ( #3236 )
3 years ago
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>
3 years ago