 Glenn Jocher
|
a3ecf0fd64
|
Anchor override (#2350)
|
2021-03-02 23:08:21 -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
|
fd96810518
|
remove TTA 1 pixel offset (#2325)
|
2021-02-28 21:14:08 -08:00 |
|
|
+7
|
cbd55da5d2
|
Update yolo.py (#2120)
* Avoid mutable state in Detect
* LoadImages() pathlib update (#2140)
* Unique *.cache filenames fix (#2134)
* fix #2121
* Update test.py
* Update train.py
* Update autoanchor.py
* Update datasets.py
* Update log_dataset.py
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update train.py test batch_size (#2148)
* Update train.py
* Update loss.py
* Update train.py (#2149)
* Linear LR scheduler option (#2150)
* Linear LR scheduler option
* Update train.py
* Update data-autodownload background tasks (#2154)
* Update get_coco.sh
* Update get_voc.sh
* Update detect.py (#2167)
Without this cv2.imshow opens a window but nothing is visible
* Update requirements.txt (#2173)
* Update utils/datasets.py to support .webp files (#2174)
Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
* Changed socket port and added timeout (#2176)
* PyTorch Hub results.save('path/to/dir') (#2179)
* YOLOv5 Segmentation Dataloader Updates (#2188)
* 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
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* GitHub API rate limit fix
* update
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* astuple
* epochs
* update
* update
* ComputeLoss()
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* merge
* merge
* merge
* merge
* update
* update
* update
* update
* commit=tag == tags[-1]
* Update cudnn.benchmark
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* mosaic9
* update
* update
* update
* update
* update
* update
* institute cache versioning
* only display on existing cache
* reverse cache exists booleans
* Created using Colaboratory
* 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>
* 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>
* LoadStreams() frame loss bug fix (#2222)
* Update yolo.py channel array (#2223)
* Add check_imshow() (#2231)
* Add check_imshow()
* Update general.py
* Update general.py
* Update CI badge (#2230)
* Add isdocker() (#2232)
* Add isdocker()
* Update general.py
* Update general.py
* YOLOv5 Hub URL inference bug fix (#2250)
* Update common.py
* Update common.py
* Update common.py
* Improved hubconf.py CI tests (#2251)
* Unified hub and detect.py box and labels plotting (#2243)
* reset head
* Update inference default to multi_label=False (#2252)
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
* Robust objectness loss balancing (#2256)
* Created using Colaboratory
* Update minimum stride to 32 (#2266)
* Dynamic ONNX engine generation (#2208)
* add: dynamic onnx export
* delete: test onnx inference
* fix dynamic output axis
* Code reduction
* fix: dynamic output axes, dynamic input naming
* Remove fixed axes
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update greetings.yml for auto-rebase on PR (#2272)
* Update Dockerfile with apt install zip (#2274)
* FLOPS min stride 32 (#2276)
Signed-off-by: xiaowo1996 <429740343@qq.com>
* Update README.md
* Amazon AWS EC2 startup and re-startup scripts (#2185)
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
* Amazon AWS EC2 startup and re-startup scripts (#2282)
* Update train.py (#2290)
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
* Improved model+EMA checkpointing (#2292)
* Enhanced model+EMA checkpointing
* update
* bug fix
* bug fix 2
* always save optimizer
* ema half
* remove model.float()
* model half
* carry ema/model in fp32
* rm model.float()
* both to float always
* cleanup
* cleanup
* Improved model+EMA checkpointing 2 (#2295)
* Fix labels being missed when image extension appears twice in filename (#2300)
* W&B entity support (#2298)
* W&B entity support
* shorten wandb_entity to entity
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Avoid mutable state in Detect
* Update yolo and remove .to(device)
Co-authored-by: Oleg Boiko <oboiko@chegg.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: train255 <thanhdd.it@gmail.com>
Co-authored-by: ab-101 <56578530+ab-101@users.noreply.github.com>
Co-authored-by: Transigent <wbdsmith@optusnet.com.au>
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
Co-authored-by: Daniel Khromov <danielkhromov@gmail.com>
Co-authored-by: VdLMV <jaap@vdlmv.nl>
Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Yann Defretin <kinoute@gmail.com>
Co-authored-by: Aditya Lohia <64709773+aditya-dl@users.noreply.github.com>
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: xiaowo1996 <429740343@qq.com>
Co-authored-by: Iden Craven <iden.craven@gmail.com>
Co-authored-by: Jan Hajek <toretak@users.noreply.github.com>
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
|
2021-02-26 15:07:40 -08:00 |
|
|
|
95aefea493
|
Dynamic ONNX engine generation (#2208)
* add: dynamic onnx export
* delete: test onnx inference
* fix dynamic output axis
* Code reduction
* fix: dynamic output axes, dynamic input naming
* Remove fixed axes
Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2021-02-21 21:50:44 -08:00 |
|
|
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 |
|
|
|
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 KhromovandGlenn Jocher
|
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 |
|
|
|
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 |
|
|
NanoCode012andGlenn Jocher
|
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 |
|