Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity
1,273 Commits 1 Branch 0 Tags
20d45aa4f18d7df7148fc6dd69a9c0607003f004
Commit Graph
465 Commits
Author SHA1 Message Date
ZigarssandGlenn Jocher 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>
 2021-06-28 13:18:45 +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 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 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
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
 2021-06-25 20:45:33 +02:00
Yonghye KwonandGlenn Jocher 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>
 2021-06-25 11:47:46 +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
Yonghye Kwon 417a2f425c Edit comment (#3759) 
edit comment
 2021-06-24 15:57:27 +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
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
 2021-06-21 14:00:25 +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
Mai Thanh MinhandGlenn Jocher 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>
 2021-06-19 11:51:21 +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 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 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
xiaowk5516andGlenn Jocher 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>
 2021-06-16 13:31:26 +02:00
Glenn Jocher 6c0e1d9fd7 Update verify_image_label() (#3635) 2021-06-16 11:12:15 +02:00
Glenn Jocher 7d3686a686 Update check_file() (#3622) 
* Update `check_file()`

* Update datasets.py
 2021-06-15 13:21:04 +02:00
Ayush ChaurasiaandGlenn Jocher 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>
 2021-06-14 18:54:58 +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 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
Ayush Chaurasia 53d4fc2e26 W&B: Allow changed in config variable #3588 2021-06-11 23:18:57 +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 095197bd4a Ignore Seaborn plot warnings (#3576) 
* Ignore Seaborn plot warnings

* Update plots.py

* Update metrics.py
 2021-06-10 13:51:29 +02:00
dependabot[bot] c6deb73a89 Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 
Bumps [pip](https://github.com/pypa/pip) from 18.1 to 19.2.
- [Release notes](https://github.com/pypa/pip/releases)
- [Changelog](https://github.com/pypa/pip/blob/main/NEWS.rst)
- [Commits](https://github.com/pypa/pip/compare/18.1...19.2)

---
updated-dependencies:
- dependency-name: pip
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-06-09 21:14:09 +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
Glenn Jocher 8b5086c21b Multi-GPU default to single device 0 (#3554) 
* Multi-GPU default to single device 0

* Multi-GPU default to single device 0

* add space
 2021-06-09 15:31:47 +02:00
Glenn Jocher 958ab92dc1 Remove opt from create_dataloader()` (#3552) 2021-06-09 13:14:56 +02:00
Glenn Jocher f8ec71e1c2 Update check_python(minimum=3.6.2) (#3548) 2021-06-09 11:34:15 +02:00
Glenn Jocher 1b5edb6f8e Update dataset_stats() for HUB (#3536) 
* Update `dataset_stats()` for HUB 

Cleanup of b6fdd2e

* autodownload flag

* Update general.py

* cleanup
 2021-06-09 10:56:11 +02:00
Glenn Jocher b6fdd2e5e5 Create dataset_stats() for HUB 2021-06-08 23:09:45 +02:00
Glenn Jocher 8d52c1c5c5 Update datasets.py (#3531) 
Minor updates to https://github.com/ultralytics/yolov5/pull/3505, inplace accumulation.
 2021-06-08 18:36:40 +02:00
Dean MarkandGlenn Jocher 28bff22df8 Use multi-threading in cache_labels (#3505) 
* Use multi threading in cache_labels

* PEP8 reformat

* Add num_threads

* changed ThreadPool.imap_unordered to Pool.imap_unordered

* Remove inplace additions

* Update datasets.py

refactor initial desc

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-06-08 18:00:21 +02:00
Glenn Jocher 317f2ccc9d Improved check_requirements() offline-handling (#3466) 
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
 2021-06-04 21:34:36 +02:00
Glenn Jocher 563ea9475a Add check_git_status() 5 second timeout (#3464) 
* Add check_git_status() 5 second timeout

This should prevent the SSH Git bug that we were discussing @KalenMike

* cleanup

* replace timeout with check_output built-in timeout
 2021-06-04 21:17:36 +02:00
Yonghye Kwon c37f072ba7 Faster HSV augmentation (#3462) 
remove datatype conversion process that can be skipped
 2021-06-04 20:02:20 +02:00
Glenn Jocher d40481acc5 Add timeout() class (#3460) 
* Add `timeout()` class

* rearrange order
 2021-06-04 15:23:33 +02:00
Glenn Jocher 8e3b4a0bf3 Update MixUp augmentation alpha=beta=32.0 (#3455) 
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
 2021-06-04 12:47:53 +02:00
Glenn Jocher f8651c388f Parse URL authentication (#3424) 
* Parse URL authentication

* urllib.parse.unquote()

* improved error handling

* improved error handling

* remove %3F

* update check_file()
 2021-06-03 18:44:58 +02:00
chocosajandBuildTools 3cb9ad4fc4 Update FLOPs description (#3422) 
* Update README.md

* Changing FLOPS to FLOPs.

Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
 2021-06-03 12:31:51 +02:00
Glenn Jocher fdbe527dc0 Revert "cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379)" (#3395) 
This reverts commit 21a9607e00.
 2021-05-31 10:39:00 +02:00
Glenn Jocher d833ab3d25 Create is_pip() function (#3391) 
Returns `True` if file is part of pip package. Useful for contextual behavior modification.

```python
def is_pip():
    # Is file in a pip package?
    return 'site-packages' in Path(__file__).absolute().parts
```
 2021-05-30 20:52:42 +02:00
tudouleiandGlenn Jocher 21a9607e00 cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379) 
* Update datasets.py

* comment

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-29 21:12:01 +02:00
Glenn Jocher ba6f3f974b Enable direct --weights URL definition (#3373) 
* Enable direct `--weights URL` definition

@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```

* cleanup

* bug fixes

* weights = attempt_download(weights)

* Update experimental.py

* Update hubconf.py

* return bug fix

* comment mirror

* min_bytes
 2021-05-28 15:18:44 +02:00
Glenn Jocher 4d4a2b0520 Ignore blank lines in *.txt labels (#3366) 
Fix for https://github.com/ultralytics/yolov5/issues/958#issuecomment-849512083
 2021-05-27 14:31:26 +02:00