• Glenn Jocher's avatar
    New YOLOv5 Classification Models (#8956) · d3ea0df8
    Glenn Jocher 提交于
    * Update
    
    * Logger step fix: Increment step with epochs (#8654)
    
    * enhance
    
    * revert
    
    * allow training from scratch
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Update --img argument from train.py 
    
    single line
    
    * fix image size from 640 to 128
    
    * suport custom dataloader and augmentation
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * format
    
    * Update dataloaders.py
    
    * Single line return, single line comment, remove unused argument
    
    * address PR comments
    
    * fix spelling
    
    * don't augment eval set
    
    * use fstring
    
    * update augmentations.py
    
    * new maning convention for transforms
    
    * reverse if statement, inline ops
    
    * reverse if statement, inline ops
    
    * updates
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * update dataloaders
    
    * Remove additional if statement
    
    * Remove is_train as redundant
    
    * Cleanup
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Cleanup2
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Update classifier.py
    
    * Update augmentations.py
    
    * fix: imshow clip warning
    
    * update
    
    * Revert ToTensorV2 removal
    
    * Update classifier.py
    
    * Update normalize values, revert uint8
    
    * normalize image using cv2
    
    * remove dedundant comment
    
    * Update classifier.py
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * replace print with logger
    
    * commit steps
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ciCo-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
    Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
    
    * 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
    
    * Allow logging models from GenericLogger (#8676)
    
    * enhance
    
    * revert
    
    * allow training from scratch
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Update --img argument from train.py 
    
    single line
    
    * fix image size from 640 to 128
    
    * suport custom dataloader and augmentation
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * format
    
    * Update dataloaders.py
    
    * Single line return, single line comment, remove unused argument
    
    * address PR comments
    
    * fix spelling
    
    * don't augment eval set
    
    * use fstring
    
    * update augmentations.py
    
    * new maning convention for transforms
    
    * reverse if statement, inline ops
    
    * reverse if statement, inline ops
    
    * updates
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * update dataloaders
    
    * Remove additional if statement
    
    * Remove is_train as redundant
    
    * Cleanup
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Cleanup2
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Update classifier.py
    
    * Update augmentations.py
    
    * fix: imshow clip warning
    
    * update
    
    * Revert ToTensorV2 removal
    
    * Update classifier.py
    
    * Update normalize values, revert uint8
    
    * normalize image using cv2
    
    * remove dedundant comment
    
    * Update classifier.py
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * replace print with logger
    
    * commit steps
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * support final model logging
    
    * update
    
    * update
    
    * update
    
    * update
    
    * remove curses
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Update classifier.py
    
    * Update __init__.py
    Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
    Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update dataset download
    
    * Update dataset download
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Pass imgsz to classify_transforms()
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Cos scheduler
    
    * Cos scheduler
    
    * Remove unused args
    
    * Update
    
    * Add seed
    
    * Add seed
    
    * Update
    
    * Update
    
    * Add run(), main()
    
    * Merge master
    
    * Merge master
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Update
    
    * Create YOLOv5 BaseModel class (#8829)
    
    * Create BaseModel
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * fix
    
    * Hub load device fix
    
    * Update
    Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
    
    * Add experiment
    
    * Merge master
    
    * Attach names
    
    * weight decay = 1e-4
    
    * weight decay = 5e-5
    
    * update smart_optimizer console printout
    
    * fashion-mnist fix
    
    * Merge master
    
    * Update Table
    
    * Update Table
    
    * Remove destroy process group
    
    * add kwargs to forward()
    
    * fuse fix for resnet50
    
    * nc, names fix for resnet50
    
    * nc, names fix for resnet50
    
    * ONNX CPU inference fix
    
    * revert
    
    * cuda
    
    * if augment or visualize
    
    * if augment or visualize
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * New smart_inference_mode()
    
    * Update README
    
    * Refactor into /classify dir
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * reset defaults
    
    * reset defaults
    
    * fix gpu predict
    
    * warmup
    
    * ema half fix
    
    * spacing
    
    * remove data
    
    * remove cache
    
    * remove denormalize
    
    * save run settings
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * verbose false on initial plots
    
    * new save_yaml() function
    
    * Update ci-testing.yml
    
    * Path(data) CI fix
    
    * Separate classification CI
    
    * fix val
    
    * fix val
    
    * fix val
    
    * smartCrossEntropyLoss
    
    * skip validation on hub load
    
    * autodownload with working dir root
    
    * str(data)
    
    * Dataset usage example
    
    * im_show normalize
    
    * im_show normalize
    
    * add imagenet simple names to multibackend
    
    * Add validation speeds
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * 24-space names
    
    * Update bash scripts
    
    * Update permissions
    
    * Add bash script arguments
    
    * remove verbose
    
    * TRT data fix
    
    * names generator fix
    
    * optimize if names
    
    * update usage
    
    * Add local loading
    
    * Verbose=False
    
    * update names printing
    
    * Add Usage examples
    
    * Add Usage examples
    
    * Add Usage examples
    
    * Add Usage examples
    
    * named_children
    
    * reshape_classifier_outputs
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * update
    
    * update
    
    * fix CI
    
    * fix incorrect class substitution
    
    * fix incorrect class substitution
    
    * remove denormalize
    
    * ravel fix
    
    * cleanup
    
    * update opt file printing
    
    * update opt file printing
    
    * update defaults
    
    * add opt to checkpoint
    
    * Add warning
    
    * Add comment
    
    * plot half bug fix
    
    * Use NotImplementedError
    
    * fix export shape report
    
    * Fix TRT load
    
    * cleanup CI
    
    * profile comment
    
    * CI fix
    
    * Add cls models
    
    * avoid inplace error
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Fix usage examples
    
    * Update README
    
    * [pre-commit.ci] auto fixes from pre-commit.com hooks
    
    for more information, see https://pre-commit.ci
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    
    * Update README
    Co-authored-by: 's avatarAyush Chaurasia <ayush.chaurarsia@gmail.com>
    Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
    d3ea0df8
experimental.py 4.1 KB