Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,360 Commits 1 Branch 0 Tags
d8f18834a246cfe3589406635c7e990f8043130a
Commit Graph
507 Commits
Author SHA1 Message Date
Glenn Jocher d8f18834a2 Update profile() for CUDA Memory allocation (#4239) 
* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Update profile()

* Cleanup
 2021-07-30 18:17:19 +02:00
Glenn Jocher 9468657502 Update profiler (#4236) 2021-07-30 16:00:47 +02:00
Glenn Jocher 7820614c40 Add @try_except decorator (#4224) 2021-07-29 17:23:35 +02:00
Glenn Jocher b60b62e874 PyCharm reformat (#4209) 
* PyCharm reformat

* YAML reformat

* Markdown reformat
 2021-07-28 23:35:14 +02:00
Ayush ChaurasiaandGlenn Jocher 750465edae W&B: More improvements and refactoring (#4205) 
* Improve docstrings and run names

* default wandb login prompt with timeout

* return key

* Update api_key check logic

* Properly support zipped dataset feature

* update docstring

* Revert tuorial change

* extend changes to log_dataset

* add run name

* bug fix

* bug fix

* Update comment

* fix import check

* remove unused import

* Hardcore .yaml file extension

* reduce code

* Reformat using pycharm

* Remove redundant try catch

* More refactoring and bug fixes

* retry

* Reformat using pycharm

* respect LOGGERS include list

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-07-28 23:25:15 +02:00
Ayush ChaurasiaandGlenn Jocher e88e8f7a98 W&B: Restructure code to support the new dataset_check() feature (#4197) 
* Improve docstrings and run names

* default wandb login prompt with timeout

* return key

* Update api_key check logic

* Properly support zipped dataset feature

* update docstring

* Revert tuorial change

* extend changes to log_dataset

* add run name

* bug fix

* bug fix

* Update comment

* fix import check

* remove unused import

* Hardcore .yaml file extension

* reduce code

* Reformat using pycharm

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-07-28 17:40:08 +02:00
Glenn Jocher 5d66e48723 Train from --data path/to/dataset.zip feature (#4185) 
* Train from `--data path/to/dataset.zip` feature

* Update dataset_stats()

* cleanup

* cleanup2
 2021-07-28 02:04:10 +02:00
Ayush Chaurasia 63a1971e31 Improve docstrings and run names (#4174) 2021-07-27 15:13:32 +02:00
Glenn Jocher 0ad6301c96 Update script headers (#4163) 
* Update download script headers

* cleanup

* bug fix attempt

* bug fix attempt2

* bug fix attempt3

* cleanup
 2021-07-26 15:23:33 +02:00
Glenn Jocher 96e36a7c91 New CSV Logger (#4148) 
* New CSV Logger

* cleanup

* move batch plots into Logger

* rename comment

* Remove total loss from progress bar

* mloss :-1 bug fix

* Update plot_results()

* Update plot_results()

* plot_results bug fix
 2021-07-25 19:06:37 +02:00
Glenn Jocher efe60b5681 Refactor train.py and val.py loggers (#4137) 
* Update loggers

* Config

* Update val.py

* cleanup

* fix1

* fix2

* fix3 and reformat

* format sweep.py

* Logger() class

* cleanup

* cleanup2

* wandb package import fix

* wandb package import fix2

* txt fix

* fix4

* fix5

* fix6

* drop wandb into utils/loggers

* fix 7

* rename loggers/wandb_logging to loggers/wandb

* Update message

* Update message

* Update message

* cleanup

* Fix x axis bug

* fix rank 0 issue

* cleanup
 2021-07-25 01:18:39 +02:00
Glenn Jocher 63dd65e7ed Update train.py (#4136) 
* Refactor train.py

* Update imports

* Update imports

* Update optimizer

* cleanup
 2021-07-24 16:11:39 +02:00
imyhxy 4495e00016 Fix indentation in log_training_progress() (#4126) 2021-07-23 14:55:00 +02:00
Glenn Jocher 4bad914761 Update resume.py (#4115) 2021-07-22 17:22:11 +02:00
Glenn Jocher 2c073cd207 Add train.py `--img-size floor (#4099) 2021-07-21 16:50:47 +02:00
Ayush Chaurasia 0cc7c58787 W&B: fix refactor bugs (#4069) 2021-07-19 13:57:13 +02:00
Glenn Jocher 7fdcc77bf4 Albumentations >= 1.0.3 (#4068) 2021-07-19 13:23:19 +02:00
Glenn Jocher c8a98cb7cb Missing nc and names handling in check_dataset() (#4066) 2021-07-19 13:10:21 +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
Ayush ChaurasiaandGlenn Jocher 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>
 2021-07-14 19:13:02 +02:00
Glenn Jocher 720aaa65c8 Rename test.py to val.py (#4000) 2021-07-14 15:43:54 +02:00
Glenn Jocher b3dabdcc38 Update probability to p (#3980) 2021-07-12 15:54:43 +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 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 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 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
ketan-bandGlenn Jocher 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>
 2021-07-04 12:55:57 +02:00
Valentin AliferovandGlenn Jocher 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>
 2021-07-02 13:25:54 +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
Feras OughaliandGlenn Jocher 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>
 2021-06-30 12:11:29 +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
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