Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
928 Commits 1 Branch 0 Tags
ca5b10b759d2e41221e7ffddcefe1f8087791dec
Commit Graph
928 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher ca5b10b759 Update train.py (#2290) 
* Update train.py

* Update train.py

* Update train.py

* Update train.py

* Create train.py
 2021-02-24 13:31:20 -08:00
Glenn Jocher 0070995bd5 Amazon AWS EC2 startup and re-startup scripts (#2282) 2021-02-24 01:43:59 -08:00
Glenn Jocher d5d275b6e9 Amazon AWS EC2 startup and re-startup scripts (#2185) 
* Amazon AWS EC2 startup and re-startup scripts

* Create resume.py

* cleanup
 2021-02-23 23:10:14 -08:00
Glenn Jocher 7a6870b81f Update README.md 2021-02-23 11:27:44 -08:00
xiaowo1996 83dc1b4484 FLOPS min stride 32 (#2276) 
Signed-off-by: xiaowo1996 <429740343@qq.com>
 2021-02-23 09:38:56 -08:00
Glenn Jocher cc79f3a9ea Update Dockerfile with apt install zip (#2274) 2021-02-22 22:50:00 -08:00
Glenn Jocher 32dd1614f4 Update greetings.yml for auto-rebase on PR (#2272) 2021-02-22 18:34:07 -08:00
Aditya LohiaShivam SwanrkarGlenn Jocher
 		95aefea493 Dynamic ONNX engine generation (#2208) 
* add: dynamic onnx export

* delete: test onnx inference

* fix dynamic output axis

* Code reduction

* fix: dynamic output axes, dynamic input naming

* Remove fixed axes

Co-authored-by: Shivam Swanrkar <ss8464@nyu.edu>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-21 21:50:44 -08:00
Glenn Jocher e27ca0d845 Update minimum stride to 32 (#2266) 2021-02-21 21:46:42 -08:00
Glenn Jocher 095d2c11d8 Created using Colaboratory 2021-02-20 13:21:58 -08:00
Glenn Jocher 6f5d6fcdaa Robust objectness loss balancing (#2256) 2021-02-20 11:19:01 -08:00
Glenn Jocher c09964c27c Update inference default to multi_label=False (#2252) 
* Update inference default to multi_label=False

* bug fix

* Update plots.py

* Update plots.py
 2021-02-19 15:39:09 -08:00
Glenn Jocher ab2da5edc8 Merge remote-tracking branch 'origin/master' 2021-02-19 15:21:04 -08:00
Glenn Jocher 47faf95079 reset head 2021-02-19 15:20:41 -08:00
Yann Defretin 5f42643a53 Unified hub and detect.py box and labels plotting (#2243) 2021-02-19 12:38:05 -08:00
Glenn Jocher db28ce61ac Improved hubconf.py CI tests (#2251) 2021-02-19 12:35:38 -08:00
Glenn Jocher 9d87307784 YOLOv5 Hub URL inference bug fix (#2250) 
* Update common.py

* Update common.py

* Update common.py
 2021-02-19 11:22:42 -08:00
Glenn Jocher d2e754b67b Add isdocker() (#2232) 
* Add isdocker()

* Update general.py

* Update general.py
 2021-02-16 15:58:07 -08:00
Glenn Jocher 5a40ce65ce Update CI badge (#2230) 2021-02-16 15:27:24 -08:00
Glenn Jocher 26c2e54c8f Add check_imshow() (#2231) 
* Add check_imshow()

* Update general.py

* Update general.py
 2021-02-16 13:56:47 -08:00
Glenn Jocher f8464b4f66 Update yolo.py channel array (#2223) 2021-02-15 21:21:53 -08:00
Glenn Jocher 7b833e37bf LoadStreams() frame loss bug fix (#2222) 2021-02-15 11:02:20 -08:00
VdLMVJaap van de LoosdrechtGlenn Jocher
 		3ff783c18f TTA augument boxes one pixel shifted in de-flip ud and lr (#2219) 
* TTA augument boxes one pixel shifted in de-flip ud and lr

* PEP8 reformat

Co-authored-by: Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-15 10:49:22 -08:00
Daniel KhromovandGlenn Jocher 3e560e2fae YOLOv5 PyTorch Hub results.save() method retains filenames (#2194) 
* save results with name

* debug

* save original imgs names

* Update common.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-12 15:37:51 -08:00
Glenn Jocher 17ac94b796 Created using Colaboratory 2021-02-11 22:39:37 -08:00
Glenn Jocher bdd88e1ed7 YOLOv5 Segmentation Dataloader Updates (#2188) 
* Update C3 module

* Update C3 module

* Update C3 module

* Update C3 module

* update

* update

* update

* update

* update

* update

* update

* update

* update

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update datasets

* update

* update

* update

* update attempt_downlaod()

* merge

* merge

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* parameterize eps

* comments

* gs-multiple

* update

* max_nms implemented

* Create one_cycle() function

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* GitHub API rate limit fix

* update

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* ComputeLoss

* astuple

* epochs

* update

* update

* ComputeLoss()

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* merge

* merge

* merge

* merge

* update

* update

* update

* update

* commit=tag == tags[-1]

* Update cudnn.benchmark

* update

* update

* update

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* update

* update

* update

* update

* update

* mosaic9

* update

* update

* update

* update

* update

* update

* institute cache versioning

* only display on existing cache

* reverse cache exists booleans
 2021-02-11 21:22:45 -08:00
Glenn Jocher 404749a33c PyTorch Hub results.save('path/to/dir') (#2179) 2021-02-10 16:10:43 -08:00
NanoCode012 a5d5f9262d Changed socket port and added timeout (#2176) 2021-02-10 13:01:48 -08:00
Transigent c9bda112ae Update utils/datasets.py to support .webp files (#2174) 
Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
 2021-02-10 00:16:38 -08:00
Glenn Jocher ace3e02e40 Update requirements.txt (#2173) 2021-02-09 22:03:29 -08:00
ab-101 c32b0aff76 Update detect.py (#2167) 
Without this cv2.imshow opens a window but nothing is visible
 2021-02-08 23:13:40 -08:00
Glenn Jocher a5359f6c32 Update data-autodownload background tasks (#2154) 
* Update get_coco.sh

* Update get_voc.sh
 2021-02-06 22:13:39 -08:00
Glenn Jocher 6b634c6b87 Linear LR scheduler option (#2150) 
* Linear LR scheduler option

* Update train.py
 2021-02-06 11:26:54 -08:00
Glenn Jocher ad839eda38 Update train.py (#2149) 2021-02-06 11:21:04 -08:00
Glenn Jocher 86897e3663 Update train.py test batch_size (#2148) 
* Update train.py

* Update loss.py
 2021-02-06 10:29:32 -08:00
train255andGlenn Jocher e9b3de463a Unique *.cache filenames fix (#2134) 
* fix #2121

* Update test.py

* Update train.py

* Update autoanchor.py

* Update datasets.py

* Update log_dataset.py

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-05 11:12:43 -08:00
Glenn Jocher 4bdc5a397e LoadImages() pathlib update (#2140) 2021-02-05 09:06:23 -08:00
Ayush ChaurasiaandGlenn Jocher 73a0669930 Start setup for improved W&B integration (#1948) 
* Add helper functions for wandb and artifacts

* cleanup

* Reorganize files

* Update wandb_utils.py

* Update log_dataset.py

We can remove this code, as the giou hyp has been deprecated for a while now.

* Reorganize and update dataloader call

* yaml.SafeLoader

* PEP8 reformat

* remove redundant checks

* Add helper functions for wandb and artifacts

* cleanup

* Reorganize files

* Update wandb_utils.py

* Update log_dataset.py

We can remove this code, as the giou hyp has been deprecated for a while now.

* Reorganize and update dataloader call

* yaml.SafeLoader

* PEP8 reformat

* remove redundant checks

* Update util files

* Update wandb_utils.py

* Remove word size

* Change path of labels.zip

* remove unused imports

* remove --rect

* log_dataset.py cleanup

* log_dataset.py cleanup2

* wandb_utils.py cleanup

* remove redundant id_count

* wandb_utils.py cleanup2

* rename cls

* use pathlib for zip

* rename dataloader to dataset

* Change import order

* Remove redundant code

* remove unused import

* remove unused imports

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-02-01 21:38:41 -08:00
Glenn Jocher 9646ca438a Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
Glenn Jocher be9edffded Update test.py --task speed and study (#2099) 
* Add --speed benchmark

* test range 256 - 1536

* update

* update

* update

* update
 2021-01-31 13:58:49 -08:00
Glenn Jocher a18efc3a73 Add variable-stride inference support (#2091) 2021-01-30 13:47:23 -08:00
Glenn Jocher aa02b94825 Add Kaggle badge (#2090) 2021-01-30 11:51:16 -08:00
Glenn Jocher eeb2bbf648 Add Kaggle badge (#2090) 
* Update README.md

* Update greetings.yml

* Created using Colaboratory
 2021-01-30 11:48:03 -08:00
Glenn Jocher 170d12e506 Add Amazon Deep Learning AMI environment (#2085) 
* Update greetings.yml

* Update README.md
 2021-01-29 13:49:00 -08:00
Glenn Jocher 64628d6fc5 Update tutorial.ipynb 2021-01-29 13:40:03 -08:00
Glenn Jocher 6e6f77be47 Add YOLOv5-P6 models (#2083) 2021-01-29 11:57:29 -08:00
Glenn Jocher 6bfa9c2422 GhostConv update (#2082) 2021-01-29 11:25:01 -08:00
Glenn Jocher 2acbe9699a W&B epoch logging update (#2073) 2021-01-28 18:57:24 -08:00
Glenn Jocher 630ec06af6 Update to colors.TABLEAU_COLORS (#2069) 2021-01-28 12:37:21 -08:00
Glenn Jocher f639e14e4d Metric-Confidence plots feature addition (#2057) 
* Metric-Confidence plots feature addition

* cleanup

* Metric-Confidence plots feature addition

* cleanup

* Update run-once lines

* cleanup

* save all 4 curves to wandb
 2021-01-27 17:10:53 -08:00