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
3 vuotta sitten
Glenn Jocher
a3ecf0fd64
Anchor override ( #2350 )
3 vuotta sitten
Glenn Jocher
fd96810518
remove TTA 1 pixel offset ( #2325 )
3 vuotta sitten
oleg
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>
3 vuotta sitten
Glenn Jocher
f8464b4f66
Update yolo.py channel array ( #2223 )
3 vuotta sitten
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>
3 vuotta sitten
Glenn Jocher
6bfa9c2422
GhostConv update ( #2082 )
3 vuotta sitten
Glenn Jocher
08d3119e09
Add histogram equalization fcn ( #2049 )
3 vuotta sitten
Abhiram V
4501169547
Update yolo.py with yaml.SafeLoader ( #1970 )
3 vuotta sitten
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
3 vuotta sitten
Glenn Jocher
9922c1183f
leaf Variable inplace bug fix ( #1759 )
3 vuotta sitten
Glenn Jocher
394d1c89f3
Input channel yaml['ch'] addition ( #1741 )
3 vuotta sitten
Glenn Jocher
799724108f
Update C3 module ( #1705 )
3 vuotta sitten
Glenn Jocher
d929bb656c
Implement default class names ( #1609 )
3 vuotta sitten
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
4 vuotta sitten
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 vuotta sitten
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
4 vuotta sitten
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4 vuotta sitten
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4 vuotta sitten
Glenn Jocher
c2403d785c
fuse() bug fix
4 vuotta sitten
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
4 vuotta sitten
Glenn Jocher
4fce0096fa
model.add_nms() method
4 vuotta sitten
Glenn Jocher
2f77cf33f6
.fuse() additional error checking
4 vuotta sitten
Glenn Jocher
6062bb7dee
print self.yaml['nc'] override update
4 vuotta sitten
tomguluson92
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
4 vuotta sitten
Glenn Jocher
a8751e50de
model.fuse() fix for export.py ( #827 )
4 vuotta sitten
Glenn Jocher
d0f98c0095
Update yolo.py with model_info(verbose=False)
4 vuotta sitten
Glenn Jocher
b42e8a531b
optimize imports
4 vuotta sitten
Glenn Jocher
916d4aad9a
v3.0 Release ( #725 )
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
4 vuotta sitten
NanoCode012
0892c44bc4
Fix Logging ( #719 )
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 vuotta sitten
NanoCode012
4949401a94
Fix redundant outputs via Logging in DDP training ( #500 )
* Change print to logging
* Clean function set_logging
* Add line spacing
* Change leftover prints to log
* Fix scanning labels output
* Fix rank naming
* Change leftover print to logging
* Reorganized DDP variables
* Fix type error
* Make quotes consistent
* Fix spelling
* Clean function call
* Add line spacing
* Update datasets.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 vuotta sitten
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
4 vuotta sitten
Glenn Jocher
d989bc9260
remove NBSP
4 vuotta sitten
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
4 vuotta sitten
Glenn Jocher
91af0401d8
PyTorch 1.6.0 compatibility updates
4 vuotta sitten
Glenn Jocher
1d17b9af0f
update yolo.py TTA flexibility and extensibility ( #506 )
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
4 vuotta sitten
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 vuotta sitten
Glenn Jocher
2b1892430f
yolo.py bug fix
4 vuotta sitten
Glenn Jocher
02445d176d
improved model.yaml source tracking
4 vuotta sitten
Glenn Jocher
04bdbe4104
fuse update
4 vuotta sitten
Glenn Jocher
38f5c1ad1d
pruning and sparsity initial commit
4 vuotta sitten
Glenn Jocher
997ba7b346
import yaml in yolo.py
4 vuotta sitten
Glenn Jocher
4e2d24602d
update yolo.py
4 vuotta sitten
Glenn Jocher
bb3c346916
model.yaml nc inherited from dataset.yaml
4 vuotta sitten
Glenn Jocher
655895a838
module updates
4 vuotta sitten
Glenn Jocher
9a9c4f1259
update yolo.py default --cfg
4 vuotta sitten
Glenn Jocher
8699c319ca
update yolo.py
4 vuotta sitten
Glenn Jocher
53294f28b4
update yolo.py
4 vuotta sitten
Glenn Jocher
fc171e2678
check_anchor_order() update
4 vuotta sitten
Glenn Jocher
80b82e8bde
update yolo.py
4 vuotta sitten