Glenn Jocher
f2d97ebb25
Remove DDP MultiHeadAttention fix ( #3768 )
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
03281f8c76
COCO annotations JSON fix ( #3764 )
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
ae4261c774
Force non-zero hyp evolution weights `w` ( #3748 )
Fix for https://github.com/ultralytics/yolov5/issues/3741
3 years ago
Glenn Jocher
fdc22398fa
Create `data/hyps` directory ( #3747 )
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
fcakyon
0e2d0d54d7
Fix typo ( #3729 )
3 years ago
Glenn Jocher
b83e1a4adc
Fix `img2label_paths()` order ( #3720 )
* Fix `img2label_paths()` order
* fix, 1
3 years ago
Glenn Jocher
1f69d12591
Update 4 main ops for paths and .run() ( #3715 )
* Add yolov5/ to path
* rename functions to run()
* cleanup
* rename fix
* CI fix
* cleanup find models/export.py
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
e8810a53e8
Update DDP backend `if dist.is_nccl_available()` ( #3705 )
3 years ago
Glenn Jocher
fbf41e0913
Add `train.run()` method ( #3700 )
* Update train.py explicit arguments
* Update train.py
* Add run method
3 years ago
Glenn Jocher
c1af67dcd4
Add torch DP warning ( #3698 )
3 years ago
Glenn Jocher
b3e2f4e08d
Eliminate `total_batch_size` variable ( #3697 )
* Eliminate `total_batch_size` variable
* cleanup
* Update train.py
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
Glenn Jocher
5bab9a28e4
Reformat ( #3694 )
3 years ago
lb-desupervised
bfb2276b1d
Slightly modify CLI execution ( #3687 )
* Slightly modify CLI execution
This simple change makes it easier to run the primary functions of this
repo (train/detect/test) from within Python. An object which represents
`opt` can be constructed and fed to the `main` function of each of these
modules, rather than having to call the lower level functions directly,
or run the module as a script.
* Update export.py
Add CLI parsing update for more convenient module usage within Python.
Co-authored-by: Lewis Belcher <lb@desupervised.io>
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
463628a4d8
Update README.md ( #3684 )
3 years ago
ZouJiu1
2729761458
EMA changes for pre-model's batch_size ( #3681 )
* EMA changes for pre-model's batch_size
* Update train.py
* Update torch_utils.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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
2296f1546f
Update `WORLD_SIZE` and `RANK` retrieval ( #3670 )
3 years ago
Glenn Jocher
045d5d8629
Update TensorBoard ( #3669 )
3 years ago
Glenn Jocher
fa201f968e
Update `train(hyp, *args)` to accept `hyp` file or dict ( #3668 )
3 years ago
Glenn Jocher
6d6e2ca65f
Update train.py ( #3667 )
3 years ago
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
3 years ago
Glenn Jocher
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* 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
SpongeBab
df7706d802
Update README.md ( #3650 )
Be more user-friendly to new users
3 years ago
Mai Thanh Minh
fa29125f18
`is_coco` list fix ( #3646 )
3 years ago
Glenn Jocher
6187edcb53
Update CONTRIBUTING.md ( #3647 )
3 years ago
Glenn Jocher
3ce0db89b0
Update CONTRIBUTING.md ( #3645 )
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
* Update CONTRIBUTING.md
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
bb79e13d52
RUN pip install --no-cache -U torch torchvision ( #3637 )
3 years ago
Glenn Jocher
6c0e1d9fd7
Update `verify_image_label()` ( #3635 )
3 years ago
Glenn Jocher
65f81bfefa
Add `**/*.torchscript.pt` ( #3634 )
3 years ago
Glenn Jocher
de56813ba8
FROM nvcr.io/nvidia/pytorch:21.05-py3 ( #3633 )
3 years ago
Glenn Jocher
30e4c4f092
Update README.md ( #3624 )
3 years ago
Glenn Jocher
7d3686a686
Update `check_file()` ( #3622 )
* Update `check_file()`
* Update datasets.py
3 years ago
Wei Quan
4c5d9bff80
Fix incorrect end epoch comment ( #3612 )
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
masood azhar
6062319ec3
Update datasets.py ( #3591 )
* 'changes-in_dataset'
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
f8adee156d
merge develop
3 years ago
Kalen Michael
3133607560
Simplify README.md ( #3530 )
* Update README.md
* added hosted images
* added new logo
* testing image hosting
* changed svgs to pngs
* removed old header
* Update README.md
* correct colab image source
* splash.jpg
* rocket and W&B fix
* added contributing template
* added social media to top section
* increased size of top social media
* cleanup and updates
* rearrange quickstarts
* API cleanup
* PyTorch Hub cleanup
* Add tutorials
* cleanup
* update CONTRIBUTING.md
* Update README.md
* update wandb link
* Update README.md
* remove tutorials header
* update environments and integrations
* Comment API image
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* double spaces after section
* Update README.md
* Update README.md
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
88b1945241
Delete __init__.py ( #3596 )
3 years ago