Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,321 Commits 1 Branch 0 Tags
b1be6850050959a09c3e26813646c52b2a73b1a0
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher b1be685005 Module super().__init__() (#4065) 
* Module `super().__init__()`

* remove NMS
 2021-07-19 12:41:15 +02:00
Glenn JocherandAyush Chaurasia f7d8562060 val.py refactor (#4053) 
* val.py refactor

* cleanup

* cleanup

* cleanup

* cleanup

* save after eval

* opt.imgsz bug fix

* wandb refactor

* dataloader to train_loader

* capitalize global variables

* runs/hub/exp to runs/detect/exp

* refactor wandb logging

* Refactor wandb operations (#4061)

Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
 2021-07-19 10:43:01 +02:00
Glenn Jocher 9dd33fd20f AutoShape PosixPath support (#4047) 
* AutoShape PosixPath support

Usage example:

````python
from pathlib import Path

model = ...
file = Path('data/images/zidane.jpg')

results = model(file)
```

* Update common.py
 2021-07-18 15:25:37 +02:00
Glenn Jocher dd62e2d05c Update README.md (#4041) 
* Update README.md

* Update README.md

* Update README.md
 2021-07-17 23:12:27 +02:00
Glenn Jocher 0067d9578a Update greetings.yml (#4037) 2021-07-17 15:47:05 +02:00
Glenn Jocher 951922c735 Add --sync-bn known issue (#4032) 
* Add `--sync-bn` known issue

* Update train.py
 2021-07-17 13:07:19 +02:00
Glenn Jocher b7e985e397 Update greetings.yml (#4024) 
* Update greetings.yml

* Update greetings.yml
 2021-07-16 14:38:59 +02:00
Glenn Jocher 720aaa65c8 Rename test.py to val.py (#4000) 2021-07-14 15:43:54 +02:00
Glenn Jocher 8ee9fd1505 Update README.md (#3996) 2021-07-13 23:07:09 +02:00
Glenn Jocher d204a61834 Alert (no detections) (#3984) 
* `Detections()` class `print()` overload

* Update common.py
 2021-07-12 19:48:47 +02:00
Glenn Jocher b3dabdcc38 Update probability to p (#3980) 2021-07-12 15:54:43 +02:00
Glenn Jocher 41fdf9fa53 Created using Colaboratory 2021-07-12 12:43:26 +02:00
Glenn Jocher a544d59f52 Update requirements.txt albumentations>=1.0.2 (#3972) 2021-07-11 19:07:42 +02:00
Glenn Jocher 90e60b403d Update np.random.random() to random.random() (#3967) 2021-07-11 16:56:36 +02:00
Glenn Jocher 8298ce5e88 Update albumentations>=1.0.2 (#3966) 2021-07-11 15:49:30 +02:00
Glenn Jocher 80299a57e2 Numerical stability fix for Albumentations (#3958) 2021-07-10 19:50:53 +02:00
Glenn Jocher 443af8b25a Cache v0.4 update (#3954) 2021-07-10 14:18:46 +02:00
Glenn Jocher a26e7de2bf Update augmentations.py (#3948) 2021-07-09 16:45:04 +02:00
Glenn Jocher 248504cf13 Feature visualization improvements 32 (#3947) 2021-07-09 15:23:02 +02:00
Glenn Jocher dabad5793a Created using Colaboratory 2021-07-08 15:45:53 +02:00
Glenn Jocher 8c6f9e15bf Update dataset_stats() for zipped datasets (#3926) 
* Update `dataset_stats()` for zipped datasets

@KalenMike

* cleanup
 2021-07-08 11:42:30 +02:00
Glenn Jocher 850970e081 feature_visualization() CUDA fix (#3925) 2021-07-07 16:23:31 +02:00
Glenn Jocher 411842e058 Fix torch.hub.list('ultralytics/yolov5') pathlib bug (#3921) 2021-07-07 16:08:42 +02:00
Glenn Jocher 87b094bcbc Feature visualization update (#3920) 
* Feature visualization update

* Save to jpg (faster)

* Save to png
 2021-07-07 15:41:58 +02:00
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
 2021-07-05 18:01:54 +02:00
Glenn Jocher 6a3ee7cf03 Hub models map_location=device (#3894) 
* Hub models `map_location=device`

* cleanup
 2021-07-05 16:20:46 +02:00
Glenn Jocher 8930e22cce Evolution commented hyp['anchors'] fix (#3887) 
Fix for `KeyError: 'anchors'` error when start hyperparameter evolution:
```bash
python train.py --evolve
```

```bash
Traceback (most recent call last):
  File "E:\yolov5\train.py", line 623, in <module>
    hyp[k] = max(hyp[k], v[1])  # lower limit
KeyError: 'anchors'
```
 2021-07-05 12:48:27 +02:00
Glenn Jocher 3c3f8fbd5d Improved BGR2RGB speeds (#3880) 
* Update BGR2RGB ops

* speed improvements

* cleanup
 2021-07-04 20:12:32 +02:00
Glenn Jocher 9e8fb9fd0b Create utils/augmentations.py (#3877) 
* Create `utils/augmentations.py`

* cleanup
 2021-07-04 18:14:04 +02:00
Glenn Jocher 81b31824f5 Models *.yaml reformat (#3875) 2021-07-04 16:55:08 +02:00
Glenn Jocher bd88e7f4f2 Created using Colaboratory 2021-07-04 15:55:23 +02:00
Glenn Jocher 4717a3b038 Created using Colaboratory 2021-07-02 12:24:27 +02:00
Glenn Jocher b6863385b5 Created using Colaboratory 2021-07-02 12:23:09 +02:00
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
 2021-07-01 00:35:04 +02:00
Glenn Jocher 25d1f2932c Plot AutoShape() detections in ascending order (#3843) 2021-06-30 15:10:40 +02:00
Glenn Jocher 57c5d02bbe Concise TransformerBlock() (#3821) 2021-06-29 16:03:10 +02:00
Glenn Jocher 5ea771d93d Move IoU functions to metrics.py (#3820) 2021-06-29 13:18:13 +02:00
Glenn Jocher 3213d8713f Fix for dataset_stats() with updated data.yaml (#3819) 
@KalenMike
 2021-06-29 12:44:59 +02:00
Glenn Jocher 02719dde52 Update feature_visualization() (#3807) 
* Update `feature_visualization()`

Only plot for data with height, width > 1

* cleanup

* Cleanup
 2021-06-28 13:48:14 +02:00
Glenn Jocher 47543f97b1 FROM nvcr.io/nvidia/pytorch:21.05-py3 (#3794) 2021-06-26 17:12:43 +02:00
Glenn Jocher 5e6886c860 Update Dockerfile (#3792) 2021-06-26 16:51:17 +02:00
Glenn Jocher 96c87f1711 nvcr.io/nvidia/pytorch:21.06-py3 (#3791) 2021-06-26 16:28:06 +02:00
Glenn Jocher 07166ba38c Update Usage examples (#3790) 2021-06-26 16:09:56 +02:00
Glenn Jocher 92d49fde35 Update seeds for single-GPU reproducibility (#3789) 
For seed=0 on single-GPU.
 2021-06-26 15:42:40 +02:00
Glenn Jocher 8e7f285051 Fix datasets for aws and get_coco.sh (#3788) 
* merge master

* Update get_coco.sh
 2021-06-26 15:29:16 +02:00
Glenn Jocher bd581b330b Create hyp.scratch-p6.yaml (#3787) 2021-06-26 15:13:06 +02:00
Glenn Jocher 8035b61682 Update objectness IoU sort (#3786) 2021-06-26 14:52:18 +02:00
Glenn JocherandU-LAPTOP-5N89P8V7\banhu 157aa2f886 Objectness IoU Sort (#3610) 
Co-authored-by: U-LAPTOP-5N89P8V7\banhu <ban.huang@foxmail.com>
 2021-06-26 14:45:53 +02:00
Glenn Jocher 9dc5d35fce Update README.md fix banner width (#3785) 2021-06-26 13:05:46 +02:00
Glenn Jocher f89941711c NGA xView 2018 Dataset Auto-Download (#3775) 
* update clip_coords for numpy

* uncomment

* cleanup

* Add autosplits

* fix

* cleanup
 2021-06-26 00:49:05 +02:00
Glenn Jocher f2d97ebb25 Remove DDP MultiHeadAttention fix (#3768) 2021-06-25 12:52:05 +02:00
Glenn Jocher 03281f8c76 COCO annotations JSON fix (#3764) 2021-06-25 01:55:53 +02:00
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
 2021-06-25 01:25:03 +02:00
Glenn Jocher ae4261c774 Force non-zero hyp evolution weights w (#3748) 
Fix for https://github.com/ultralytics/yolov5/issues/3741
 2021-06-23 12:56:22 +02:00
Glenn Jocher fdc22398fa Create data/hyps directory (#3747) 2021-06-23 12:49:38 +02:00
Glenn Jocher 63060910a6 Update check_datasets() for dynamic unzip path (#3732) 
@KalenMike
 2021-06-22 16:05:38 +02:00
Glenn Jocher 9ac7d388a9 Backwards compatible cache version checks (#3730) 2021-06-22 13:50:47 +02:00
Glenn Jocher b83e1a4adc Fix img2label_paths() order (#3720) 
* Fix `img2label_paths()` order

* fix, 1
 2021-06-21 22:50:56 +02:00
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
 2021-06-21 17:25:04 +02:00
Glenn Jocher e8810a53e8 Update DDP backend if dist.is_nccl_available() (#3705) 2021-06-20 17:15:42 +02:00
Glenn Jocher fbf41e0913 Add train.run() method (#3700) 
* Update train.py explicit arguments

* Update train.py

* Add run method
 2021-06-20 15:06:58 +02:00
Glenn Jocher c1af67dcd4 Add torch DP warning (#3698) 2021-06-19 19:50:46 +02:00
Glenn Jocher b3e2f4e08d Eliminate total_batch_size variable (#3697) 
* Eliminate `total_batch_size` variable

* cleanup

* Update train.py
 2021-06-19 19:14:59 +02:00
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
 2021-06-19 16:30:25 +02:00
Glenn Jocher 5bab9a28e4 Reformat (#3694) 2021-06-19 14:21:28 +02:00
Glenn Jocher 814806c61d Update cache check (#3691) 
Swapped order of operations for faster first per https://github.com/ultralytics/yolov5/commit/f527704cd32c42bc0bba9cce04601783b8563204#r52362419
 2021-06-19 11:22:09 +02:00
Glenn Jocher 463628a4d8 Update README.md (#3684) 2021-06-18 17:12:42 +02:00
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
 2021-06-18 10:21:47 +02:00
Glenn Jocher 2296f1546f Update WORLD_SIZE and RANK retrieval (#3670) 2021-06-17 23:24:30 +02:00
Glenn Jocher 045d5d8629 Update TensorBoard (#3669) 2021-06-17 22:12:42 +02:00
Glenn Jocher fa201f968e Update train(hyp, *args) to accept hyp file or dict (#3668) 2021-06-17 22:03:25 +02:00
Glenn Jocher 6d6e2ca65f Update train.py (#3667) 2021-06-17 21:32:39 +02:00
Glenn Jocher ac34834563 Update download_weights.sh comment (#3662) 2021-06-17 16:21:10 +02:00
Glenn Jocher 2754adad46 Remove /weights directory (#3659) 
* Remove `/weights` directory

* cleanup
 2021-06-17 14:44:12 +02:00
Glenn Jocher 9b6dba6207 Update dataset_stats() to list of dicts (#3657) 
* Update `dataset_stats()` to list of dicts

@KalenMike

* Update datasets.py
 2021-06-17 13:59:52 +02:00
Glenn Jocher 6187edcb53 Update CONTRIBUTING.md (#3647) 2021-06-16 19:57:11 +02:00
Glenn Jocher 3ce0db89b0 Update CONTRIBUTING.md (#3645) 
* Update CONTRIBUTING.md

* Update CONTRIBUTING.md

* Update CONTRIBUTING.md

* Update CONTRIBUTING.md
 2021-06-16 19:48:30 +02:00
Glenn Jocher bb79e13d52 RUN pip install --no-cache -U torch torchvision (#3637) 2021-06-16 12:35:33 +02:00
Glenn Jocher 6c0e1d9fd7 Update verify_image_label() (#3635) 2021-06-16 11:12:15 +02:00
Glenn Jocher 65f81bfefa Add **/*.torchscript.pt (#3634) 2021-06-16 10:49:50 +02:00
Glenn Jocher de56813ba8 FROM nvcr.io/nvidia/pytorch:21.05-py3 (#3633) 2021-06-16 10:40:31 +02:00
Glenn Jocher 30e4c4f092 Update README.md (#3624) 2021-06-15 15:44:10 +02:00
Glenn Jocher 7d3686a686 Update check_file() (#3622) 
* Update `check_file()`

* Update datasets.py
 2021-06-15 13:21:04 +02:00
Glenn Jocher 239a11c197 Download COCO and VOC by default (#3608) 2021-06-14 13:58:54 +02:00
Glenn Jocher f8adee156d merge develop 2021-06-14 10:45:15 +02:00
Glenn Jocher 88b1945241 Delete __init__.py (#3596) 2021-06-12 15:21:37 +02:00
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)
```
 2021-06-12 13:26:41 +02:00
Glenn Jocher 4984cf54be train.py GPU memory fix (#3590) 
* train.py GPU memory fix

* ema

* cuda

* cuda

* zeros input

* to device

* batch index 0
 2021-06-11 20:24:03 +02:00
Glenn Jocher e8c5237403 ConfusionMatrix normalize=True fix (#3587) 2021-06-11 11:46:05 +02:00
Glenn Jocher ec2da4a82c Add ConfusionMatrix normalize=True flag (#3586) 2021-06-11 11:37:08 +02:00
Glenn Jocher 46e1fdfbc6 Update stale.yml (#3585) 2021-06-11 11:23:39 +02:00
Glenn Jocher 5c32bd3080 Created using Colaboratory 2021-06-10 16:01:34 +02:00
Glenn Jocher 53ed872c28 Update export.py, yolo.py sys.path.append() (#3579) 2021-06-10 15:35:22 +02:00
Glenn Jocher 095197bd4a Ignore Seaborn plot warnings (#3576) 
* Ignore Seaborn plot warnings

* Update plots.py

* Update metrics.py
 2021-06-10 13:51:29 +02:00
Glenn Jocher 4695ca8314 Refactoring cleanup (#3565) 
* Refactoring cleanup

* Update test.py
 2021-06-09 22:50:27 +02:00
Glenn Jocher 0e5cfdbea7 Refactor models/export.py arguments (#3564) 
* Refactor models/export.py arguments

* cleanup

* cleanup
 2021-06-09 22:43:46 +02:00
Glenn Jocher 66cf5c28c1 Refactor detect.py arguments (#3559) 
* Refactor detect.py arguments

@SkalskiP @KalenMike

* unused ok

* comment arguments
 2021-06-09 22:19:34 +02:00
Glenn Jocher a9553c04a7 Refactor test.py arguments (#3558) 
* remove opt from test()

* pass kwargs

* update comments

* revert accidental default change

* multiple --img options

* add comments
 2021-06-09 21:36:10 +02:00
Glenn Jocher 4ef9261870 Remove redundant speed/study half argument (#3557) 2021-06-09 16:30:01 +02:00
Glenn Jocher 5948f20a3d Update test.py profiling (#3555) 
* Update test.py profiling

* half_precision to half

* inplace
 2021-06-09 16:25:17 +02:00