mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
998 Commits
10 Branches
10 Tags
584 MiB
Commit Graph

37 Commits (866bc7d640b04913943820c45636e7c2da6d8245)

Author SHA1 Message Date
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
 2021-03-29 15:19:07 +02:00
Glenn Jocher ee169834bd
PyTorch Hub custom model to CUDA device fix (#2636) 
Fix for #2630 raised by @Pro100rus32
 2021-03-28 17:22:00 +02:00
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
 2021-03-24 16:23:54 +01:00
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
 2021-03-24 15:42:00 +01:00
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
 2021-03-14 23:16:17 -07:00
Glenn Jocher 569757ecc0
Add autoShape() speed profiling (#2459) 
* Add autoShape() speed profiling

* Update common.py

* Create README.md

* Update hubconf.py

* cleanuip
 2021-03-13 19:50:34 -08:00
Glenn Jocher fab5085674
EMA bug fix 2 (#2330) 
* EMA bug fix 2

* update
 2021-03-01 17:13:34 -08:00
Glenn Jocher db28ce61ac
Improved hubconf.py CI tests (#2251) 2021-02-19 12:35:38 -08:00
Glenn Jocher 9a3da79b4a
Update requirements.txt (#2021) 
* Update requirements.txt

* Update ci-testing.yml

* Update hubconf.py
 2021-01-23 12:51:04 -08:00
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>
 2020-12-26 19:58:26 -08:00
Glenn Jocher e92245a38c
Simplified PyTorch hub for custom models (#1677) 2020-12-12 11:42:27 -08:00
Glenn Jocher f7a923b446
Simplified PyTorch hub for custom models (#1677) 2020-12-12 11:37:53 -08:00
Glenn Jocher 87ca35b922
Simplified PyTorch hub for custom models (#1677) 2020-12-12 11:16:57 -08:00
Glenn Jocher fa8f1fb0e9
Simplify autoshape() post-process (#1653) 
* Simplify autoshape() post-process

* cleanup

* cleanup
 2020-12-09 07:44:06 -08:00
Glenn Jocher 311de00855 PyTorch Hub and autoShape update (#1415) 2020-11-16 23:20:17 +01:00
Glenn Jocher 2aaaec3cca PyTorch Hub and autoShape update (#1415) 2020-11-16 23:16:35 +01:00
Glenn Jocher 7816c0c2d3 PyTorch Hub and autoShape update (#1415) 2020-11-16 23:15:43 +01:00
Glenn Jocher 93fc607f07 PyTorch Hub and autoShape update (#1415) 2020-11-16 23:14:07 +01:00
Glenn Jocher f5429260ca
PyTorch Hub and autoShape update (#1415) 
* PyTorch Hub and autoShape update

* comment x for imgs

* reduce comment
 2020-11-16 23:09:55 +01:00
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>
 2020-11-10 13:01:19 +01:00
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
 2020-11-08 19:39:05 +01:00
Glenn Jocher c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates (#1233) 
* torch 1.7.0 compatibility updates

* add inference verification
 2020-10-28 15:03:50 +01:00
Glenn Jocher 7f1640695b
Update hubconf.py (#1210) 2020-10-26 01:06:54 +01:00
Glenn Jocher 0fda95aaf4 Update cache out-of-date warning 2020-10-23 15:13:42 +02:00
Glenn Jocher 3b57cb5641
Simplified inference (#1153) 2020-10-15 20:10:08 +02:00
Glenn Jocher 4fce0096fa model.add_nms() method 2020-09-20 16:31:40 -07:00
Glenn Jocher c4cb78570c add NMS to pretrained pytorch hub models 2020-09-18 18:17:11 -07:00
Jirka Borovec d5b6416c87
Explicit Imports (#498) 
* expand imports

* optimize

* miss

* fix
 2020-08-02 15:47:36 -07:00
Glenn Jocher 84a946615a update hubconf.py 2020-07-16 16:44:00 -07:00
Glenn Jocher a5818b280e expired pytorch hub cache bug fix #365 2020-07-11 15:04:53 -07:00
Glenn Jocher 6e3c3b62b2 expired pytorch hub cache bug fix #365 2020-07-11 15:03:12 -07:00
Glenn Jocher 6fb5ff014c FP16 to FP32 ckpt load 2020-06-15 13:21:34 -07:00
Glenn Jocher 4f6c0cf4c5
Update hubconf.py 2020-06-14 12:15:14 -07:00
Lijun Yu 0dec8ffd66 path fix for torch hub 2020-06-14 14:47:40 -04:00
Glenn Jocher 18b4da91e0 PyTorch Hub update 2020-06-10 19:14:47 -07:00
Glenn Jocher c8f0694894 PyTorch Hub update 2020-06-10 19:13:04 -07:00
Glenn Jocher a814720403 PyTorch Hub updates 2020-06-10 19:11:11 -07:00