mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,148 Commits 10 Branches 10 Tags
Commit Graph

184 Commits

Author SHA1 Message Date
Glenn Jocher
c09964c27c
Update inference default to multi_label=False (#2252) 
* Update inference default to multi_label=False

* bug fix

* Update plots.py

* Update plots.py
 2021-02-19 15:39:09 -08:00
Yann Defretin
5f42643a53
Unified hub and detect.py box and labels plotting (#2243) 2021-02-19 12:38:05 -08:00
Glenn Jocher
db28ce61ac
Improved hubconf.py CI tests (#2251) 2021-02-19 12:35:38 -08:00
Glenn Jocher
9d87307784
YOLOv5 Hub URL inference bug fix (#2250) 
* Update common.py

* Update common.py

* Update common.py
 2021-02-19 11:22:42 -08:00
Glenn Jocher
f8464b4f66
Update yolo.py channel array (#2223) 2021-02-15 21:21:53 -08:00
VdLMV
3ff783c18f
TTA augument boxes one pixel shifted in de-flip ud and lr (#2219) 
* TTA augument boxes one pixel shifted in de-flip ud and lr

* PEP8 reformat

Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-15 10:49:22 -08:00
Daniel Khromov
3e560e2fae
YOLOv5 PyTorch Hub results.save() method retains filenames (#2194) 
* save results with name

* debug

* save original imgs names

* Update common.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-12 15:37:51 -08:00
Glenn Jocher
404749a33c
PyTorch Hub results.save('path/to/dir') (#2179) 2021-02-10 16:10:43 -08:00
Glenn Jocher
9646ca438a
Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
Glenn Jocher
6e6f77be47
Add YOLOv5-P6 models (#2083) 2021-01-29 11:57:29 -08:00
Glenn Jocher
6bfa9c2422
GhostConv update (#2082) 2021-01-29 11:25:01 -08:00
Glenn Jocher
08d3119e09
Add histogram equalization fcn (#2049) 2021-01-26 17:30:42 -08:00
Glenn Jocher
2fc4760257
Update autoshape .print() and .save() (#2022) 2021-01-22 16:38:06 -08:00
Glenn Jocher
aac33f87ad
Update inference multiple-counting (#2019) 
* Update inference multiple-counting

* update github check
 2021-01-22 15:39:08 -08:00
Abhiram V
4501169547
Update yolo.py with yaml.SafeLoader (#1970) 2021-01-18 10:47:00 -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
1d1c0567a4
PyTorch Hub results.render() (#1897) 2021-01-10 20:06:11 -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
0b6266f5e0
Update yolov3-tiny.yaml (#1823) 2021-01-02 13:01:05 -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
9922c1183f
leaf Variable inplace bug fix (#1759) 2020-12-22 17:27:40 -08:00
Glenn Jocher
394d1c89f3
Input channel yaml['ch'] addition (#1741) 2020-12-19 10:54:01 -08:00
Glenn Jocher
6bd5e8bca7
nn.SiLU() export support (#1713) 2020-12-16 17:55:57 -08:00
Glenn Jocher
799724108f
Update C3 module (#1705) 2020-12-15 22:13:08 -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
791dadb51c
Pycocotools best.pt after COCO train (#1616) 
* Pycocotools best.pt after COCO train

* cleanup
 2020-12-06 14:58:33 +01:00
Glenn Jocher
d929bb656c
Implement default class names (#1609) 2020-12-06 12:41:37 +01:00
Glenn Jocher
ba48f867ea
Add bias to Classify() (#1601) 2020-12-04 15:06:33 +01:00
Glenn Jocher
199c9c7874
Detection() device bug fix (#1455) 2020-11-19 13:45:36 +01:00
Glenn Jocher
225845e781
Update test.py for IoU in native image-space (#1439) 
* Update test.py for IoU in native image-space

* remove redundant

* gn to device

* remove output scale_coords

* --img-size correction

* update

* native-space labels

* pred to predn

* remove clip_coords()
 2020-11-18 22:50:21 +01:00
Glenn Jocher
97c6a59dd1
PANet anchor order correction (#1420) 2020-11-17 10:13:00 +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
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation (#1398) 
* Improved FLOPS computation

* update comment
 2020-11-14 14:39:46 +01:00
Glenn Jocher
fe341fa44d
Utils reorganization (#1392) 
* Utils reorganization

* Add new utils files

* cleanup

* simplify

* reduce datasets.py

* remove evolve.sh

* loadWebcam cleanup
 2020-11-14 11:50:32 +01:00
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup (#1325) 
* Create flatten_recursive() helper function

* cleanup

* print torch version
 2020-11-09 12:24:11 +01:00
Glenn Jocher
15a10609fe
autoshape() update for PIL greyscale inputs (#1279) 
* autoshape update for PIL greyscale inputs

* autoshape update for PIL greyscale inputs
 2020-11-04 14:20:11 +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
Yi_Zhang
f532bc68a9
GhostConv() bug fix (#1176) 2020-10-20 13:42:11 +02:00
Glenn Jocher
3b57cb5641
Simplified inference (#1153) 2020-10-15 20:10:08 +02:00
Glenn Jocher
d45e349a1e
Rearrange export input after checks (#1118) 
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
 2020-10-11 16:23:36 +02:00
Glenn Jocher
c2403d785c
fuse() bug fix 2020-10-11 00:42:54 +02:00
Glenn Jocher
883924d9dc classifier, export, torch seed updates 2020-10-06 14:54:02 +02:00
Glenn Jocher
a12698f4c9
Update ci-testing.yml reflecting sys.path additions (#869) 
* Update ci-testing.yml

* Update yolo.py

* Update ci-testing.yml

* Update export.py

* Update yolo.py
 2020-10-04 18:50:32 +02:00
PetrDvoracek
d55c326dda
export.py folder sys.path addition (#1052) 2020-10-04 17:42:05 +02:00
Glenn Jocher
4fce0096fa model.add_nms() method 2020-09-20 16:31:40 -07:00
Glenn Jocher
2f77cf33f6 .fuse() additional error checking 2020-09-20 12:04:20 -07:00
Glenn Jocher
c4cb78570c add NMS to pretrained pytorch hub models 2020-09-18 18:17:11 -07:00
Glenn Jocher
6062bb7dee print self.yaml['nc'] override update 2020-09-11 17:44:11 -07:00
tomguluson92
35dc935504
remove nn.Conv2d in default settings (#949) 2020-09-11 16:35:48 -07:00
Glenn Jocher
d68179953d Update export.py 2020-09-02 13:23:29 -07:00