• oleg's avatar
    Update yolo.py (#2120) · cbd55da5
    oleg 提交于
    * 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: 's avatarGlenn 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: 's avatarGlenn 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: 's avatarJaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
    Co-authored-by: 's avatarGlenn 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: 's avatarShivam Swanrkar <ss8464@nyu.edu>
    Co-authored-by: 's avatarGlenn 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: 's avatarxiaowo1996 <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: 's avatarJan Hajek <jan.hajek@gmail.com>
    Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
    
    * Avoid mutable state in Detect
    
    * Update yolo and remove .to(device)
    Co-authored-by: 's avatarOleg Boiko <oboiko@chegg.com>
    Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
    Co-authored-by: 's avatartrain255 <thanhdd.it@gmail.com>
    Co-authored-by: 's avatarab-101 <56578530+ab-101@users.noreply.github.com>
    Co-authored-by: 's avatarTransigent <wbdsmith@optusnet.com.au>
    Co-authored-by: 's avatarNanoCode012 <kevinvong@rocketmail.com>
    Co-authored-by: 's avatarDaniel Khromov <danielkhromov@gmail.com>
    Co-authored-by: 's avatarVdLMV <jaap@vdlmv.nl>
    Co-authored-by: 's avatarJaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
    Co-authored-by: 's avatarYann Defretin <kinoute@gmail.com>
    Co-authored-by: 's avatarAditya Lohia <64709773+aditya-dl@users.noreply.github.com>
    Co-authored-by: 's avatarShivam Swanrkar <ss8464@nyu.edu>
    Co-authored-by: 's avatarxiaowo1996 <429740343@qq.com>
    Co-authored-by: 's avatarIden Craven <iden.craven@gmail.com>
    Co-authored-by: 's avatarJan Hajek <toretak@users.noreply.github.com>
    Co-authored-by: 's avatarJan Hajek <jan.hajek@gmail.com>
    cbd55da5
yolo.py 11.6 KB