1aea74cddb
Add new --vid-stride inference parameter for videos ( #9256 )
...
* fps feature/skip frame added
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* predict.py updates
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* remove unused attribute
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Cleanup
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update predict.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update detect.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Rename skip_frame to vid_stride
* cleanup
* cleanup2
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-04 17:15:53 +02:00
Glenn Jocher
ffdb58b0e0
Move cudnn.benchmarks(True) to LoadStreams ( #9258 )
...
* Move cudnn.benchmarks(True) to LoadStreams
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Move cudnn.benchmarks(True) to LoadStreams
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-02 14:12:10 +02:00
Glenn Jocher
9da6d0f9f5
Add LoadImages._cv2_rotate() ( #9249 )
...
Optional manual rotation code per iPhone rotation issue in https://github.com/ultralytics/yolov5/issues/8493
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-01 22:30:26 +02:00
Glenn Jocher
c91d1db716
Update dataloaders.py ( #9250 )
...
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-09-01 14:30:21 +02:00
spacewalk01
79e181a83b
Add support for *.pfm images ( #9230 )
...
add support for pfm image
2022-08-31 17:31:13 +02:00
Yannick Merkli and Yannick Merkli
5f1000a499
Remove usage of pathlib.Path.unlink(missing_ok=...) ( #9227 )
...
remove usage of pathlib.Path.unlink(missing_ok=...)
Co-authored-by: Yannick Merkli <ymerkli@latticeflow.ai >
2022-08-30 21:57:36 +02:00
Glenn Jocher and pre-commit-ci[bot]
6e7a7ae7ed
New LetterBox(size) CenterCrop(size), ToTensor() transforms ( #9213 )
...
* New LetterBox transform
YOLOv5 LetterBox class for image preprocessing, i.e. T.Compose([T.ToTensor(), LetterBox(size)])
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update augmentations.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update augmentations.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update augmentations.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-30 15:17:58 +02:00
Glenn Jocher
5fb267f3e5
Duplicate segment verification fix ( #9225 )
...
Solution by @Laughing-q to resolve duplicate segment verification bug in https://github.com/ultralytics/yolov5/pull/9052#issuecomment-1231426638
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-08-30 13:40:51 +02:00
Glenn Jocher
48e56d3c9b
Add optional transforms argument to LoadStreams() ( #9105 )
...
* Add optional `transforms` argument to LoadStreams()
Prepare for streaming classification support
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Cleanup
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* fix
* batch size > 1 fix
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-08-23 14:37:46 +02:00
Glenn Jocher
4a8ab3bc42
VOC names dictionary fix ( #9034 )
...
* VOC names dictionary fix
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
* Update dataloaders.py
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-08-19 15:07:04 +02:00
64e0757edf
[Classify]: Allow inference on dirs and videos ( #9003 )
...
* allow image dirs
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update predict.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
* Update predict.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update predict.py
* Update predict.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-08-17 22:41:43 +02:00
Glenn Jocher
e83b422a69
Update dataset names from array to dictionary ( #9000 )
...
* Migrate dataset names to dictionary
* fix check
* backwards compat
* predict fix
* val fix
* Keep dataset stats behavior identical
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-08-17 17:52:53 +02:00
d3ea0df8b9
New YOLOv5 Classification Models ( #8956 )
...
* 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.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn 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: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn 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: pre-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: Ayush Chaurasia <ayush.chaurarsia@gmail.com >
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-08-17 11:59:01 +02:00
Glenn Jocher
bc9fcb1767
HUBDatasetStats() preview images to 50 quality ( #8880 )
...
@kalenmike should represent a 30% filesize reduction vs 75 quality
2022-08-05 14:45:41 +02:00
Glenn Jocher
84e7748564
Update dataloaders.py remove float64 shapes ( #8865 )
...
May help https://github.com/ultralytics/yolov5/issues/8862
2022-08-04 17:06:08 +02:00
Glenn Jocher and pre-commit-ci[bot]
b367860196
New HUBDatasetStats() class ( #8716 )
...
* New `HUBDatasetStats()` class
Usage examples:
```
from utils.dataloaders import *
stats = HUBDatasetStats('coco128.yaml', autodownload=True) # method 1
stats = HUBDatasetStats('path/to/coco128_with_yaml.zip') # method 1
stats.get_json(save=False)
stats.process_images()
```
@kalenmike
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-07-25 18:20:01 +02:00
Glenn Jocher
a6f197ae79
Update dataloaders.py ( #8714 )
...
* Update dataloaders.py
* Update dataloaders.py
2022-07-25 16:52:28 +02:00
Glenn Jocher
0ab303b044
Remove else: from load_image() ( #8692 )
2022-07-25 00:02:09 +02:00
1c5e92aba1
Add generator and worker seed ( #8602 )
...
* Add generator and worker seed
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
* Update dataloaders.py
* Update dataloaders.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-07-23 01:25:17 +02:00
526e650553
Fix LoadImages() with dataset YAML lists ( #8517 )
...
* Fix LoadImages with dataset yaml lists
* Update dataloaders.py
* Update dataloaders.py
* Simplify/refactor PR
* Update dataloaders.py
Co-authored-by: Colin Wong <noreply@brains4drones.com >
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2022-07-08 14:32:40 +02:00
Glenn Jocher
898332433a
Update cache comments ( #8414 )
...
* Update cache comments
For better readability
* Update dataloaders.py
2022-06-30 16:19:22 +02:00
Glenn Jocher and pre-commit-ci[bot]
d6051382f1
HUB dataset_stats() error reporting ( #8192 )
...
* HUB dataset_stats() error reporting
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update dataloaders.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-06-17 18:37:08 +02:00
Glenn Jocher
86e6add07b
Remove streaming source sleep period ( #8243 )
...
Negatively impacts YouTube inference but removes any lag on webcams/RTSP/RTMP etc.
Resolves https://github.com/ultralytics/yolov5/issues/4465
2022-06-17 17:59:27 +02:00
Glenn Jocher and pre-commit-ci[bot]
75bbaa8f44
Reject webcam inference on Colab/Kaggle ( #8209 )
...
* Reject webcam inference on Colab/Kaggle
Improve user error understanding for https://github.com/ultralytics/yolov5/issues/8180
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* cleanup
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-06-14 14:28:21 +02:00
Glenn Jocher and pre-commit-ci[bot]
c23a441c9d
Improved dataset_stats() YAML checks ( #8125 )
...
* Update dataloaders.py
* Update dataloaders.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-06-11 19:30:54 +02:00
Glenn Jocher
a9a92aec5c
cv2.resize interpolation fix (#7903 )
...
Fix for https://github.com/ultralytics/yolov5/discussions/7901
2022-05-20 12:59:05 +02:00
Glenn Jocher
f00071416f
Refactor collections and fstrings ( #7821 )
...
* Update torch_utils.py
* Additional code refactoring
* tuples to sets
* Cleanup
2022-05-15 16:38:26 +02:00
Glenn Jocher
9d8ed37df7
Rename utils/datasets.py > utils/dataloaders.py ( #7799 )
2022-05-13 14:34:16 +02:00
