+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 |
|
|
|
efa4946d15
|
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>
|
2021-02-25 19:18:19 -08:00 |
|
|
Iden Craven
|
a82dce7faa
|
Fix labels being missed when image extension appears twice in filename (#2300)
|
2021-02-25 18:05:38 -08:00 |
|
|
Glenn Jocher
|
71dd2768f2
|
Improved model+EMA checkpointing 2 (#2295)
|
2021-02-24 21:03:21 -08:00 |
|
|
Glenn Jocher
|
ec1d8496ba
|
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
|
2021-02-24 18:26:46 -08:00 |
|
|
Glenn Jocher
|
ca5b10b759
|
Update train.py (#2290)
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
|
2021-02-24 13:31:20 -08:00 |
|
|
Glenn Jocher
|
0070995bd5
|
Amazon AWS EC2 startup and re-startup scripts (#2282)
|
2021-02-24 01:43:59 -08:00 |
|
|
Glenn Jocher
|
d5d275b6e9
|
Amazon AWS EC2 startup and re-startup scripts (#2185)
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
|
2021-02-23 23:10:14 -08:00 |
|
|
Glenn Jocher
|
7a6870b81f
|
Update README.md
|
2021-02-23 11:27:44 -08:00 |
|
|
xiaowo1996
|
83dc1b4484
|
FLOPS min stride 32 (#2276)
Signed-off-by: xiaowo1996 <429740343@qq.com>
|
2021-02-23 09:38:56 -08:00 |
|
|
Glenn Jocher
|
cc79f3a9ea
|
Update Dockerfile with apt install zip (#2274)
|
2021-02-22 22:50:00 -08:00 |
|
|
Glenn Jocher
|
32dd1614f4
|
Update greetings.yml for auto-rebase on PR (#2272)
|
2021-02-22 18:34:07 -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
|
e27ca0d845
|
Update minimum stride to 32 (#2266)
|
2021-02-21 21:46:42 -08:00 |
|
|
Glenn Jocher
|
095d2c11d8
|
Created using Colaboratory
|
2021-02-20 13:21:58 -08:00 |
|
|
Glenn Jocher
|
6f5d6fcdaa
|
Robust objectness loss balancing (#2256)
|
2021-02-20 11:19:01 -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 |
|
|
Glenn Jocher
|
ab2da5edc8
|
Merge remote-tracking branch 'origin/master'
|
2021-02-19 15:21:04 -08:00 |
|
|
Glenn Jocher
|
47faf95079
|
reset head
|
2021-02-19 15:20:41 -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
|
d2e754b67b
|
Add isdocker() (#2232)
* Add isdocker()
* Update general.py
* Update general.py
|
2021-02-16 15:58:07 -08:00 |
|
|
Glenn Jocher
|
5a40ce65ce
|
Update CI badge (#2230)
|
2021-02-16 15:27:24 -08:00 |
|
|
Glenn Jocher
|
26c2e54c8f
|
Add check_imshow() (#2231)
* Add check_imshow()
* Update general.py
* Update general.py
|
2021-02-16 13:56:47 -08:00 |
|
|
Glenn Jocher
|
f8464b4f66
|
Update yolo.py channel array (#2223)
|
2021-02-15 21:21:53 -08:00 |
|
|
Glenn Jocher
|
7b833e37bf
|
LoadStreams() frame loss bug fix (#2222)
|
2021-02-15 11:02:20 -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
|
17ac94b796
|
Created using Colaboratory
|
2021-02-11 22:39:37 -08:00 |
|
|
Glenn Jocher
|
bdd88e1ed7
|
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
|
2021-02-11 21:22:45 -08:00 |
|
|
Glenn Jocher
|
404749a33c
|
PyTorch Hub results.save('path/to/dir') (#2179)
|
2021-02-10 16:10:43 -08:00 |
|
|
NanoCode012
|
a5d5f9262d
|
Changed socket port and added timeout (#2176)
|
2021-02-10 13:01:48 -08:00 |
|
|
Transigent
|
c9bda112ae
|
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.
|
2021-02-10 00:16:38 -08:00 |
|
|
Glenn Jocher
|
ace3e02e40
|
Update requirements.txt (#2173)
|
2021-02-09 22:03:29 -08:00 |
|
|
ab-101
|
c32b0aff76
|
Update detect.py (#2167)
Without this cv2.imshow opens a window but nothing is visible
|
2021-02-08 23:13:40 -08:00 |
|
|
Glenn Jocher
|
a5359f6c32
|
Update data-autodownload background tasks (#2154)
* Update get_coco.sh
* Update get_voc.sh
|
2021-02-06 22:13:39 -08:00 |
|
|
Glenn Jocher
|
6b634c6b87
|
Linear LR scheduler option (#2150)
* Linear LR scheduler option
* Update train.py
|
2021-02-06 11:26:54 -08:00 |
|
|
Glenn Jocher
|
ad839eda38
|
Update train.py (#2149)
|
2021-02-06 11:21:04 -08:00 |
|
|
Glenn Jocher
|
86897e3663
|
Update train.py test batch_size (#2148)
* Update train.py
* Update loss.py
|
2021-02-06 10:29:32 -08:00 |
|
|
train255andGlenn Jocher
|
e9b3de463a
|
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>
|
2021-02-05 11:12:43 -08:00 |
|
|
Glenn Jocher
|
4bdc5a397e
|
LoadImages() pathlib update (#2140)
|
2021-02-05 09:06:23 -08:00 |
|
|
Ayush ChaurasiaandGlenn Jocher
|
73a0669930
|
Start setup for improved W&B integration (#1948)
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Update util files
* Update wandb_utils.py
* Remove word size
* Change path of labels.zip
* remove unused imports
* remove --rect
* log_dataset.py cleanup
* log_dataset.py cleanup2
* wandb_utils.py cleanup
* remove redundant id_count
* wandb_utils.py cleanup2
* rename cls
* use pathlib for zip
* rename dataloader to dataset
* Change import order
* Remove redundant code
* remove unused import
* remove unused imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2021-02-01 21:38:41 -08:00 |
|
|
Glenn Jocher
|
9646ca438a
|
Update plot_study() (#2112)
|
2021-02-01 13:51:11 -08:00 |
|
|
Glenn Jocher
|
be9edffded
|
Update test.py --task speed and study (#2099)
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
|
2021-01-31 13:58:49 -08:00 |
|
|
Glenn Jocher
|
a18efc3a73
|
Add variable-stride inference support (#2091)
|
2021-01-30 13:47:23 -08:00 |
|
|
Glenn Jocher
|
aa02b94825
|
Add Kaggle badge (#2090)
|
2021-01-30 11:51:16 -08:00 |
|
|
Glenn Jocher
|
eeb2bbf648
|
Add Kaggle badge (#2090)
* Update README.md
* Update greetings.yml
* Created using Colaboratory
|
2021-01-30 11:48:03 -08:00 |
|
|
Glenn Jocher
|
170d12e506
|
Add Amazon Deep Learning AMI environment (#2085)
* Update greetings.yml
* Update README.md
|
2021-01-29 13:49:00 -08:00 |
|
|
Glenn Jocher
|
64628d6fc5
|
Update tutorial.ipynb
|
2021-01-29 13:40:03 -08:00 |
|