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 години
4b52e19a61
COCO evolution fix ( #3388 )
* COCO evolution fix
* cleanup
* update print
* print fix
преди 3 години
21a9607e00
`cv2.imread(img, -1)` for IMREAD_UNCHANGED ( #3379 )
* Update datasets.py
* comment
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
57f773b0ae
Update tutorial.ipynb ( #3368 )
add Open in Kaggle badge
преди 3 години
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 години
b78e30dd2b
Merge remote-tracking branch 'origin/develop' into develop
преди 3 години
3fea068384
update ci-testing.yml ( #3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
преди 3 години
bb131238aa
update ci-testing.yml ( #3322 )
* update ci-testing.yml
* update greetings.yml
* bring back os matrix
преди 3 години
4d4a2b0520
Ignore blank lines in `*.txt` labels ( #3366 )
Fix for https://github.com/ultralytics/yolov5/issues/958#issuecomment-849512083
преди 3 години
ef4d53818d
ONNX export in .train() mode fix ( #3362 )
преди 3 години
2435bfe896
Add URL download to check_file() ( #3330 )
* Add URL file download to check_file()
* cleanup
* pathlib bug fix
преди 3 години
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 години
1f8d716ec9
yolo.py header ( #3347 )
преди 3 години
aad99b63d6
TensorBoard DP/DDP graph fix ( #3325 )
преди 3 години
407dc5008e
Update README.md ( #3320 )
преди 3 години
61ea23c3fe
Implement `@torch.no_grad()` decorator ( #3312 )
* `@torch.no_grad()` decorator
* Update detect.py
преди 3 години
73a92dc1b6
Explicit `git clone` master ( #3311 )
преди 3 години
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 години
ee24ae11ea
Comment Cython ( #3297 )
Ultralytics REST API fix.
преди 3 години
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 години
0e2f2cbb51
Update LoadStreams init fallbacks ( #3295 )
преди 3 години
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 години
dd7f0b7e05
Fix TypeError: 'PosixPath' object is not iterable ( #3285 )
преди 3 години
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 години
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 години
10d56d784e
Assert `--image-weights` not combined with DDP ( #3275 )
преди 3 години
7b36e38cf8
Check CoreML models.train() mode ( #3262 )
* Check CoreML models.train() mode
* Update export.py
преди 3 години
7b863a34dc
Add Cython ( #3217 )
Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172
преди 3 години
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 години
b7cd1f540d
TensorBoard add_graph() fix ( #3236 )
преди 3 години
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 години
13a1c72699
Update datasets.py ( #3216 )
преди 3 години
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
преди 3 години
36b0a9e7d0
Add CAP_PROP_FRAME_COUNT for YouTube sources ( #3193 )
преди 3 години
ae53f50ae7
Plot labels histogram colors ( #3192 )
преди 3 години
b133baa336
Add `device` argument to PyTorch Hub models ( #3104 )
* Allow to manual selection of device for torchhub models
* single line device
nested torch.device(torch.device(device)) ok
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
9ab561dbfc
Parameterize ONNX `--opset-version` ( #3154 )
преди 3 години
be86c21c73
rename class autoShape -> AutoShape ( #3173 )
* rename class autoShape -> AutoShape
follow other class naming convention
* rename class autoShape -> AutoShape
follow other classes' naming convention
* rename class autoShape -> AutoShape
преди 3 години
17b0f71538
SKU-110K CVPR2019 Dataset Auto-Download ( #3167 )
* SKU-110K CVPR2019 Dataset Auto-Download
This PR adds the **SKU-110K CVPR2019 Dataset** https://github.com/eg4000/SKU110K_CVPR19 to the list of YOLOv5 🚀 auto-download list.
- [x] [COCO](https://cocodataset.org/#home )
- [x] [COCO128](https://www.kaggle.com/ultralytics/coco128 )
- [x] [VOC](http://host.robots.ox.ac.uk/pascal/VOC/ )
- [x] [Argoverse-HD](http://www.cs.cmu.edu/~mengtial/proj/streaming/ )
- [x] [VisDrone2019-DET](https://github.com/VisDrone/VisDrone-Dataset )
- [x] [GlobalWheat-2020](http://www.global-wheat.com/ )
- [ ] [KITTI](https://www.kaggle.com/twaldo/kitti-object-detection )
- [x] [SKU110](https://github.com/eg4000/SKU110K_CVPR19 ) (grocery store items)
- [ ] [CityScapes](https://www.cityscapes-dataset.com/ )
- [x] [Objects365](https://www.objects365.org/overview.html )
- [ ] [OpenImages](https://storage.googleapis.com/openimages/web/index.html )
```yaml
# SKU-110K retail items dataset https://github.com/eg4000/SKU110K_CVPR19
# Train command: python train.py --data SKU-110K.yaml
# Default dataset location is next to YOLOv5:
# /parent_folder
# /datasets/SKU-110K
# /yolov5
```
* Update SKU-110K.yaml
* Update SKU-110K.yaml
преди 3 години
06372b1465
Scope all hubconf.py imports for torch.hub.list() ( #3145 )
* Scope all hubconf.py imports for torch.hub.list()
* Update hubconf.py
преди 3 години
1935266951
Scope imports for torch.hub.list() improvement ( #3144 )
преди 3 години
b098600763
Update requirements.txt `onnx>=1.9.0` ( #3143 )
преди 3 години
f12cef85a3
TorchScript, ONNX, CoreML Export tutorial title ( #3142 )
преди 3 години
d9b4e6b748
Add `--include torchscript onnx coreml` argument ( #3137 )
* Allow users to skip exporting in formats that they don't care about
* Correct comments
* Update export.py
renamed --skip-format to --exclude
* Switched format from exclude to include (as instructed by @glenn-jocher )
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
bd6f6a784a
Add yolov5/__init__.py ( #3127 )
преди 3 години
7918eed040
Update https://ultralytics.com/images/zidane.jpg ( #3140 )
преди 3 години
c1c7eb023f
Update JSON response ( #3139 )
преди 3 години
52c05707d6
Update Colors() ( #3046 )
* Update Colors()
* update colors
* update colors
преди 3 години
abfcf9eb79
Replace print() with logging.info() in trainloader ( #3103 )
Might indirectly help #3095 by providing better visibility on source of corruption.
преди 3 години
25f8ab835e
detect.py streaming source `--save-crop` bug fix ( #3102 )
* detect.py streaming source --save-crop bug fix
Possible fix for #3100 .
* () parenthesis
преди 3 години
Glenn Jocher
tudoulei
Peretz Cohen
SkalskiP
Piotr Skalski
WangChaofeng
yeric1789
Charles Frye
Adrian Holovaty
Christoph Gerum
Cristi Fati
Yonghye Kwon
KC-Zhang
