19 Commits

Author SHA1 Message Date
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
Glenn Jocher
251aeafcb1
Update P5 + P6 model ensembling (#3082) 2021-05-08 19:06:12 +02:00
jluntamazon
41f5cc5637
YOLOv5 AWS Inferentia Inplace compatibility updates (#2953)
* Added flag to enable/disable all inplace and assignment operations

* Removed shape print statements

* Scope Detect/Model import to avoid circular dependency

* PEP8

* create _descale_pred()

* replace lost space

* replace list with tuple

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-30 12:54:48 +02:00
Glenn Jocher
a57f23d18b
Update detections() self.t = tuple() (#2617)
* Update detections() self.t = tuple()

Fix multiple results.print() bug.

* Update experimental.py

* Update yolo.py
2021-03-26 13:23:11 +01:00
Glenn Jocher
fab5085674
EMA bug fix 2 (#2330)
* EMA bug fix 2

* update
2021-03-01 17:13:34 -08:00
Glenn Jocher
6bfa9c2422
GhostConv update (#2082) 2021-01-29 11:25:01 -08:00
lorne
03ebe6ea14
Add nn.SiLU inplace in attempt_load() (#1940)
* sleep 3 sec to load data .

* Update train.py

* Add nn.SiLU inplace in attempt_load()

Co-authored-by: wangliang <wangliang@codingapi.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-01-15 09:58:41 -08:00
Glenn Jocher
69be8e738f
YOLOv5 v4.0 Release (#1837)
* Update C3 module

* Update C3 module

* Update C3 module

* Update C3 module

* update

* update

* update

* update

* update

* update

* update

* update

* update

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update datasets

* update

* update

* update

* update attempt_downlaod()

* merge

* merge

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* parameterize eps

* comments

* gs-multiple

* update

* max_nms implemented

* Create one_cycle() function

* update

* update

* update

* update

* update

* update

* update

* update study.png

* update study.png

* Update datasets.py
2021-01-04 19:54:09 -08:00
Glenn Jocher
799724108f
Update C3 module (#1705) 2020-12-15 22:13:08 -08: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
Yi_Zhang
f532bc68a9
GhostConv() bug fix (#1176) 2020-10-20 13:42:11 +02:00
Jirka Borovec
d5b6416c87
Explicit Imports (#498)
* expand imports

* optimize

* miss

* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher
16f6834486 update train.py and experimental.py 2020-07-08 14:23:34 -07:00
Glenn Jocher
e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07:00
Glenn Jocher
5ba1de0cdc update experimental.py with Ensemble() module 2020-07-05 15:02:56 -07:00
Glenn Jocher
655895a838 module updates 2020-07-03 13:46:12 -07:00
Glenn Jocher
b5659d1195 module updates 2020-07-01 11:44:49 -07:00
Glenn Jocher
a814720403 PyTorch Hub updates 2020-06-10 19:11:11 -07:00
Glenn Jocher
1e84a23f38 initial commit 2020-05-29 17:04:54 -07:00