Glenn Jocher
31ee54c28c
Hubconf.py bug fix ( #3007 )
3 роки тому
Glenn Jocher
7c89c829e3
cleanup
3 роки тому
Glenn Jocher
c1a44ed9c5
Update hubconf.py for unified loading ( #3005 )
3 роки тому
Glenn Jocher
d08575ee5e
PyTorch Hub load directly when possible ( #2986 )
3 роки тому
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list ( #2974 )
3 роки тому
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>
3 роки тому
Glenn Jocher
c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix ( #2831 )
* PyTorch Hub cv2 .save() .show() bug fix
cv2.rectangle() was failing on non-contiguous np array inputs. This checks for contiguous arrays and applies is necessary:
```python
imgs[i] = im if im.data.contiguous else np.ascontiguousarray(im) # update
```
* Update plots.py
```python
assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to plot_on_box() input image.'
```
* Update hubconf.py
Expand CI tests to OpenCV image.
3 роки тому
Glenn Jocher
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
3 роки тому
Glenn Jocher
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
3 роки тому
Glenn Jocher
c03d590320
Add Hub results.pandas() method ( #2725 )
* Add Hub results.pandas() method
New method converts results from torch tensors to pandas DataFrames with column names.
This PR may partially resolve issue https://github.com/ultralytics/yolov5/issues/2703
```python
results = model(imgs)
print(results.pandas().xyxy[0])
xmin ymin xmax ymax confidence class name
0 57.068970 391.770599 241.383545 905.797852 0.868964 0 person
1 667.661255 399.303589 810.000000 881.396667 0.851888 0 person
2 222.878387 414.774231 343.804474 857.825073 0.838376 0 person
3 4.205386 234.447678 803.739136 750.023376 0.658006 5 bus
4 0.000000 550.596008 76.681190 878.669922 0.450596 0 person
```
* Update comments
torch example input now shown resized to size=640 and also now a multiple of P6 stride 64 (see https://github.com/ultralytics/yolov5/issues/2722#issuecomment-814785930 )
* apply decorators
* PEP8
* Update common.py
* pd.options.display.max_columns = 10
* Update common.py
3 роки тому
Glenn Jocher
866bc7d640
Speed profiling improvements ( #2648 )
* Speed profiling improvements
* Update torch_utils.py
deepcopy() required to avoid adding elements to model.
* Update torch_utils.py
3 роки тому
Glenn Jocher
ee169834bd
PyTorch Hub custom model to CUDA device fix ( #2636 )
Fix for #2630 raised by @Pro100rus32
3 роки тому
Glenn Jocher
8ace1b1b99
YOLOv5 PyTorch Hub models >> check_requirements() ( #2588 )
* YOLOv5 PyTorch Hub models >> check_requirements()
Update YOLOv5 PyTorch Hub requirements.txt path to cache path.
* Update hubconf.py
3 роки тому
Glenn Jocher
2bcc89d762
YOLOv5 PyTorch Hub models >> check_requirements() ( #2577 )
* Update hubconf.py with check_requirements()
Dependency checks have been missing from YOLOv5 PyTorch Hub model loading, causing errors in some cases when users are attempting to import hub models in unsupported environments. This should examine the YOLOv5 requirements.txt file and pip install any missing or version-conflict packages encountered.
This is highly experimental (!), please let us know if this creates problems in your custom workflows.
* Update hubconf.py
3 роки тому
Glenn Jocher
9b11f0c58b
PyTorch Hub models default to CUDA:0 if available ( #2472 )
* PyTorch Hub models default to CUDA:0 if available
* device as string bug fix
3 роки тому
Glenn Jocher
569757ecc0
Add autoShape() speed profiling ( #2459 )
* Add autoShape() speed profiling
* Update common.py
* Create README.md
* Update hubconf.py
* cleanuip
3 роки тому
Glenn Jocher
fab5085674
EMA bug fix 2 ( #2330 )
* EMA bug fix 2
* update
3 роки тому
Glenn Jocher
db28ce61ac
Improved hubconf.py CI tests ( #2251 )
3 роки тому
Glenn Jocher
9a3da79b4a
Update requirements.txt ( #2021 )
* Update requirements.txt
* Update ci-testing.yml
* Update hubconf.py
3 роки тому
NanoCode012
14b0abe2d0
autoShape() default for PyTorch Hub models ( #1692 )
* Add autoshape parameter
* Remove autoshape call in ReadMe
* Update hubconf.py
* file/URI inputs and autoshape check passthrough
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 роки тому
Glenn Jocher
e92245a38c
Simplified PyTorch hub for custom models ( #1677 )
3 роки тому
Glenn Jocher
f7a923b446
Simplified PyTorch hub for custom models ( #1677 )
3 роки тому
Glenn Jocher
87ca35b922
Simplified PyTorch hub for custom models ( #1677 )
3 роки тому
Glenn Jocher
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
* Simplify autoshape() post-process
* cleanup
* cleanup
3 роки тому
Glenn Jocher
311de00855
PyTorch Hub and autoShape update ( #1415 )
4 роки тому
Glenn Jocher
2aaaec3cca
PyTorch Hub and autoShape update ( #1415 )
4 роки тому
Glenn Jocher
7816c0c2d3
PyTorch Hub and autoShape update ( #1415 )
4 роки тому
Glenn Jocher
93fc607f07
PyTorch Hub and autoShape update ( #1415 )
4 роки тому
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
4 роки тому
Khiem Doan
1c8464e199
Use pathlib instead of low-level module ( #1329 )
* Use pathlib instead of low-level module
* Use pathlib instead of low-level module
* Update detect.py
* Update test.py
* reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 роки тому
Glenn Jocher
4821d076e2
Increment train, test, detect runs/ ( #1322 )
* Increment train, test, detect runs/
* Update ci-testing.yml
* inference/images to data/images
* move images
* runs/exp to runs/train/exp
* update 'results saved to %s' str
4 роки тому
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4 роки тому
Glenn Jocher
7f1640695b
Update hubconf.py ( #1210 )
4 роки тому
Glenn Jocher
0fda95aaf4
Update cache out-of-date warning
4 роки тому
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4 роки тому
Glenn Jocher
4fce0096fa
model.add_nms() method
4 роки тому
Glenn Jocher
c4cb78570c
add NMS to pretrained pytorch hub models
4 роки тому
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
4 роки тому
Glenn Jocher
84a946615a
update hubconf.py
4 роки тому
Glenn Jocher
a5818b280e
expired pytorch hub cache bug fix #365
4 роки тому
Glenn Jocher
6e3c3b62b2
expired pytorch hub cache bug fix #365
4 роки тому
Glenn Jocher
6fb5ff014c
FP16 to FP32 ckpt load
4 роки тому
Glenn Jocher
4f6c0cf4c5
Update hubconf.py
4 роки тому
Lijun Yu
0dec8ffd66
path fix for torch hub
4 роки тому
Glenn Jocher
18b4da91e0
PyTorch Hub update
4 роки тому
Glenn Jocher
c8f0694894
PyTorch Hub update
4 роки тому
Glenn Jocher
a814720403
PyTorch Hub updates
4 роки тому