Ayush Chaurasia
62409eea08
W&B sweeps support ( #3938 )
* Add support for W&B Sweeps
* Update and reformat
* Update search space
* reformat
* reformat sweep.py
* Update sweep.py
* Move sweeps files to wandb dir
* Remove print
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
3 years ago
Glenn Jocher
b3dabdcc38
Update `probability` to `p` ( #3980 )
3 years ago
Glenn Jocher
90e60b403d
Update `np.random.random()` to `random.random()` ( #3967 )
3 years ago
Glenn Jocher
8298ce5e88
Update `albumentations>=1.0.2` ( #3966 )
3 years ago
Glenn Jocher
80299a57e2
Numerical stability fix for Albumentations ( #3958 )
3 years ago
Glenn Jocher
443af8b25a
Cache v0.4 update ( #3954 )
3 years ago
Glenn Jocher
a26e7de2bf
Update augmentations.py ( #3948 )
3 years ago
Glenn Jocher
248504cf13
Feature visualization improvements 32 ( #3947 )
3 years ago
Glenn Jocher
8c6f9e15bf
Update `dataset_stats()` for zipped datasets ( #3926 )
* Update `dataset_stats()` for zipped datasets
@KalenMike
* cleanup
3 years ago
Glenn Jocher
850970e081
`feature_visualization()` CUDA fix ( #3925 )
3 years ago
Glenn Jocher
87b094bcbc
Feature visualization update ( #3920 )
* Feature visualization update
* Save to jpg (faster)
* Save to png
3 years ago
Glenn Jocher
33202b7f0b
YOLOv5 + Albumentations integration ( #3882 )
* Albumentations integration
* ToGray p=0.01
* print confirmation
* create instance in dataloader init method
* improved version handling
* transform not defined fix
* assert string update
* create check_version()
* add spaces
* update class comment
3 years ago
Glenn Jocher
6a3ee7cf03
Hub models `map_location=device` ( #3894 )
* Hub models `map_location=device`
* cleanup
3 years ago
Glenn Jocher
3c3f8fbd5d
Improved BGR2RGB speeds ( #3880 )
* Update BGR2RGB ops
* speed improvements
* cleanup
3 years ago
Glenn Jocher
9e8fb9fd0b
Create `utils/augmentations.py` ( #3877 )
* Create `utils/augmentations.py`
* cleanup
3 years ago
ketan-b
9d86b54eb3
Add multi-stream saving feature ( #3864 )
* Added the recording feature for multiple streams
Thanks for the very cool repo!!
I was trying to record multiple feeds at the same time, but the current version of the detector only had one video writer and one vid_path!
So the streams were not being saved and only were initialized with one frame and this process didn't record the whole thing.
Fix:
I made a list of `vid_writer` and `vid_path` and the `i` from the loop over the `pred` took care of the writer which need to work!
I hope this helps, Thanks!
* Cleanup list lengths
* batch size variable
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Valentin Aliferov
831773f5a2
Add EXIF rotation to YOLOv5 Hub inference ( #3852 )
* rotating an image according to its exif tag
* Update common.py
* Update datasets.py
* Update datasets.py
faster
* delete extraneous gpg file
* Update common.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
c6c88dc601
Copy-Paste augmentation for YOLOv5 ( #3845 )
* Copy-paste augmentation initial commit
* if any segments
* Add obscuration rejection
* Add copy_paste hyperparameter
* Update comments
3 years ago
Feras Oughali
7d6af69638
Fix `LoadStreams()` dataloader frame skip issue ( #3833 )
* Update datasets.py to read every 4th frame of streams
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
5ea771d93d
Move IoU functions to metrics.py ( #3820 )
3 years ago
Glenn Jocher
3213d8713f
Fix for `dataset_stats()` with updated data.yaml ( #3819 )
@KalenMike
3 years ago
Glenn Jocher
02719dde52
Update `feature_visualization()` ( #3807 )
* Update `feature_visualization()`
Only plot for data with height, width > 1
* cleanup
* Cleanup
3 years ago
Zigarss
20d45aa4f1
Add feature map visualization ( #3804 )
* Add feature map visualization
Add a feature_visualization function to visualize the mid feature map of the model.
* Update yolo.py
* remove boolean from forward and reorder if statement
* remove print from forward
* General cleanup
* Indent
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
* merge master
* Update get_coco.sh
3 years ago
Glenn Jocher
8035b61682
Update objectness IoU sort ( #3786 )
3 years ago
Glenn Jocher
157aa2f886
Objectness IoU Sort ( #3610 )
Co-authored-by: U-LAPTOP-5N89P8V7\banhu <ban.huang@foxmail.com>
3 years ago
Glenn Jocher
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
3 years ago
Ayush Chaurasia
ffb6e11050
W&B: Update Tables API and comply with new dataset_check ( #3772 )
* Update tables API and windows path fix
* update dataset check
3 years ago
Yonghye Kwon
374957317a
Add `xyxy2xywhn()` ( #3765 )
* Edit Comments for numpy2torch tensor process
Edit Comments for numpy2torch tensor process
* add xyxy2xywhn
add xyxy2xywhn
* add xyxy2xywhn
* formatting
* pass arguments
pass arguments
* edit comment for xyxy2xywhn()
edit comment for xyxy2xywhn()
* cleanup datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
f79d7479da
Add optional dataset.yaml `path` attribute ( #3753 )
* Add optional dataset.yaml `path` attribute
@KalenMike
* pass locals to python scripts
* handle lists
* update coco128.yaml
* Capitalize first letter
* add test key
* finalize GlobalWheat2020.yaml
* finalize objects365.yaml
* finalize SKU-110K.yaml
* finalize SKU-110K.yaml
* finalize VisDrone.yaml
* NoneType fix
* update download comment
* voc to VOC
* update
* update VOC.yaml
* update VOC.yaml
* remove dashes
* delete get_voc.sh
* force coco and coco128 to ../datasets
* Capitalize Argoverse_HD.yaml
* Capitalize Objects365.yaml
* update Argoverse_HD.yaml
* coco segments fix
* VOC single-thread
* update Argoverse_HD.yaml
* update data_dict in test handling
* create root
3 years ago
Yonghye Kwon
417a2f425c
Edit comment ( #3759 )
edit comment
3 years ago
Glenn Jocher
63060910a6
Update `check_datasets()` for dynamic unzip path ( #3732 )
@KalenMike
3 years ago
Glenn Jocher
9ac7d388a9
Backwards compatible cache version checks ( #3730 )
3 years ago
Glenn Jocher
b83e1a4adc
Fix `img2label_paths()` order ( #3720 )
* Fix `img2label_paths()` order
* fix, 1
3 years ago
Ayush Chaurasia
75c0ff43af
[x]W&B: Don't resume transfer learning runs ( #3604 )
* Allow config cahnge
* Allow val change in wandb config
* Don't resume transfer learning runs
* Add entity in log dataset
3 years ago
Glenn Jocher
fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend ( #3680 )
* Update DDP for `torch.distributed.run`
* Add LOCAL_RANK
* remove opt.local_rank
* backend="gloo|nccl"
* print
* print
* debug
* debug
* os.getenv
* gloo
* gloo
* gloo
* cleanup
* fix getenv
* cleanup
* cleanup destroy
* try nccl
* return opt
* add --local_rank
* add timeout
* add init_method
* gloo
* move destroy
* move destroy
* move print(opt) under if RANK
* destroy only RANK 0
* move destroy inside train()
* restore destroy outside train()
* update print(opt)
* cleanup
* nccl
* gloo with 60 second timeout
* update namespace printing
3 years ago
Mai Thanh Minh
bf209f6fe9
Skip HSV augmentation when hyperparameters are [0, 0, 0] ( #3686 )
* Create shortcircuit in augment_hsv when hyperparameter are zero
* implement faster opt-in
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
814806c61d
Update cache check ( #3691 )
Swapped order of operations for faster first per f527704cd3 (r52362419)
3 years ago
Glenn Jocher
f527704cd3
Cache v0.3: improved corrupt image/label reporting ( #3676 )
* Cache v0.3: improved corrupt image/label reporting
Fix for https://github.com/ultralytics/yolov5/issues/3656#issuecomment-863660899
* cleanup
3 years ago
Glenn Jocher
9b6dba6207
Update `dataset_stats()` to list of dicts ( #3657 )
* Update `dataset_stats()` to list of dicts
@KalenMike
* Update datasets.py
3 years ago
xiaowk5516
d808855f77
Assert non-premature end of JPEG images ( #3638 )
* premature end of JPEG images
* PEP8 reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
6c0e1d9fd7
Update `verify_image_label()` ( #3635 )
3 years ago
Glenn Jocher
7d3686a686
Update `check_file()` ( #3622 )
* Update `check_file()`
* Update datasets.py
3 years ago
Ayush Chaurasia
daab682b06
Suppress wandb images size mismatch warning ( #3611 )
* supress wandb images size mismatch warning
* supress wandb images size mismatch warning
* PEP8 reformat and optimize imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
239a11c197
Download COCO and VOC by default ( #3608 )
3 years ago
Glenn Jocher
7a565f130a
Update `dataset_stats()` ( #3593 )
@KalenMike this is a PR to add image filenames and labels to our stats dictionary and to save the dictionary to JSON. Save location is next to the train labels.cache file. The single JSON contains all stats for entire dataset.
Usage example:
```python
from utils.datasets import *
dataset_stats('coco128.yaml', verbose=True)
```
3 years ago
Ayush Chaurasia
53d4fc2e26
W&B: Allow changed in config variable #3588
3 years ago
Glenn Jocher
e8c5237403
ConfusionMatrix `normalize=True` fix ( #3587 )
3 years ago
Glenn Jocher
ec2da4a82c
Add ConfusionMatrix `normalize=True` flag ( #3586 )
3 years ago