Commit Graph
100 Commits
Author SHA1 Message Date
Glenn Jocher 32dd1614f4 Update greetings.yml for auto-rebase on PR (#2272) 2021-02-22 18:34:07 -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
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
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
Glenn Jocher ace3e02e40 Update requirements.txt (#2173) 2021-02-09 22:03:29 -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
Glenn Jocher 4bdc5a397e LoadImages() pathlib update (#2140) 2021-02-05 09:06:23 -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
Glenn Jocher 2a835c79a9 Update run-once lines (#2058) 2021-01-27 16:01:24 -08:00
Glenn Jocher f59f80114c Add 'exclude' tuple to check_requirements() (#2041) 2021-01-26 21:17:36 -08:00
Glenn Jocher 59c21c7bcb W&B log epoch (#1946)
* W&B log epoch

* capitalize

* W&B log epoch

* capitalize

* Update train.py

New try using https://docs.wandb.ai/library/log#incremental-logging

* Update train.py

* Update test.py

* Update train.py

* Update plots.py

* Update train.py

* Update train.py

* label plot step -1

* update

* update

* update

* update

* update

* update

* Update train.py

* Update train.py
2021-01-26 21:16:01 -08:00
Glenn Jocher 08d3119e09 Add histogram equalization fcn (#2049) 2021-01-26 17:30:42 -08:00
Glenn Jocher 8fcbe43423 Check im.format during dataset caching (#2042)
* Check im.format during dataset caching

* Update datasets.py
2021-01-25 20:55:35 -08:00
Glenn Jocher a41d910c5d data-autodownload background tasks (#2034) 2021-01-24 18:01:58 -08:00
Glenn Jocher 3551b072b3 Docker pyYAML>=5.3.1 fix (#2031) 2021-01-24 11:54:15 -08:00
Glenn Jocher c76c607265 PyYAML==5.4.1 (#2030) 2021-01-24 10:05:40 -08:00
Glenn Jocher 9a3da79b4a Update requirements.txt (#2021)
* Update requirements.txt

* Update ci-testing.yml

* Update hubconf.py
2021-01-23 12:51:04 -08:00
Glenn Jocher 2fc4760257 Update autoshape .print() and .save() (#2022) 2021-01-22 16:38:06 -08:00
Glenn Jocher 3b7feeafdf Update general.py check_git_status() fix (#2020) 2021-01-22 15:59:01 -08:00
Glenn Jocher aac33f87ad Update inference multiple-counting (#2019)
* Update inference multiple-counting

* update github check
2021-01-22 15:39:08 -08:00
Glenn Jocher 85b75d6018 Update ci-testing.yml (#2018) 2021-01-22 15:22:54 -08:00
Glenn Jocher 77fd83225f Update google_utils.py (#2017) 2021-01-22 15:08:42 -08:00
Glenn Jocher 8dc68fcbcb Update Dockerfile (#2016) 2021-01-22 14:26:58 -08:00
Glenn Jocher 1445ab2b27 check_git_status() Windows fix (#2015)
* check_git_status() Windows fix

* Update general.py

* Update general.py

* Update general.py

* Update general.py

* Update general.py

* Update general.py
2021-01-22 12:37:12 -08:00
Glenn Jocher 046c37e465 verbose on final_epoch (#1997) 2021-01-20 18:27:38 -08:00
Glenn Jocher d9212140b3 Add xywhn2xyxy() (#1983) 2021-01-19 13:33:52 -08:00
Glenn Jocher 1ca2d26b9f Update Dockerfile (#1982) 2021-01-19 11:45:15 -08:00
Glenn Jocher b1cf25dd9a check_git_status() asserts (#1977) 2021-01-18 10:49:28 -08:00
Glenn Jocher 18c25889dc Update tutorial.ipynb 2021-01-17 13:11:28 -08:00
Glenn Jocher 3a42abd18a Created using Colaboratory 2021-01-17 13:04:16 -08:00
Glenn Jocher b26a2f6242 check_git_status() when not exist /workspace (#1966) 2021-01-17 11:55:25 -08:00
Glenn Jocher e8a41e8916 prevent check_git_status() in docker images (#1951)
* prevent check_git_status() running docker images

* Update general.py
2021-01-16 01:15:38 -08:00
Glenn Jocher ca9babb8e6 Add ComputeLoss() class (#1950) 2021-01-15 13:50:24 -08:00
Glenn Jocher f4a78e1b40 W&B mosaic log bug fix (#1949) 2021-01-15 12:09:53 -08:00
Glenn Jocher 777032efbc Daemon thread mosaic plots fix (#1943) 2021-01-14 13:16:48 -08:00
Glenn Jocher b75c432ea0 GitHub API rate limit fallback (#1930) 2021-01-13 21:09:50 -08:00
Glenn Jocher 051e9e80de check_git_status() bug fix (#1925) 2021-01-13 10:25:53 -08:00
Glenn Jocher 509dd51aca check_git_status() improvements (#1916)
* check_online()

* Update general.py

* update check_git_status()

* reverse rev-parse order

* fetch

* improved responsiveness

* comment

* comment

* remove hyp['giou'] compat warning
2021-01-12 21:51:49 -08:00
Glenn Jocher dd03b20ba5 colorstr() updates (#1909)
* W&B ImportError message fix

* colorstr() updates

* colorstr() updates

* colorstr() default to 'blue', 'bold'

* train: magenta

* train: blue
2021-01-12 10:33:15 -08:00
Glenn Jocher 1d1c0567a4 PyTorch Hub results.render() (#1897) 2021-01-10 20:06:11 -08:00
Glenn Jocher ffef77124e Update torch_utils.py (#1895) 2021-01-10 12:26:29 -08:00
Glenn Jocher 0f11aaf551 GitHub API rate limit newline (#1894) 2021-01-10 11:47:00 -08:00
Glenn Jocher d06ad3bc3d GitHub API rate limit fix (#1894) 2021-01-10 11:42:45 -08:00
Glenn Jocher 6ab589583c Add colorstr() (#1887)
* Add colorful()

* update

* newline fix

* add git description

* --always

* update loss scaling

* update loss scaling 2

* rename to colorstr()
2021-01-09 15:24:18 -08:00
Glenn Jocher 3e25f1e9e5 auto-verbose if nc <=20 (#1869) 2021-01-07 15:47:28 -08:00
Glenn Jocher fda8df7e88 actions/stale@v3 (#1868) 2021-01-07 11:25:43 -08:00
Glenn Jocher fea9c9b80c nn.SiLU() citation correction (#1713) 2021-01-06 20:58:41 -08:00
Glenn Jocher e77c77f580 Add check_requirements() (#1853)
* Add check_requirements()

* add import

* parameterize filename

* add to detect, test
2021-01-06 16:35:40 -08:00
Glenn Jocher 69be8e738f YOLOv5 v4.0 Release (#1837)
* 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 study.png

* update study.png

* Update datasets.py
2021-01-04 19:54:09 -08:00
Glenn Jocher 0e341c5660 Create one_cycle() function (#1836) 2021-01-04 15:49:08 -08:00
Glenn Jocher 9f5a18bb80 Torch CUDA synchronize update (#1826)
* torch.cuda.synchronize() update

* torch.cuda.synchronize() update

* torch.cuda.synchronize() update

* newline
2021-01-03 11:23:12 -08:00
Glenn Jocher 0b6266f5e0 Update yolov3-tiny.yaml (#1823) 2021-01-02 13:01:05 -08:00
Glenn Jocher 7180b22e2d DDP Multi-GPU --resume bug fix (#1810) 2020-12-30 12:40:45 -08:00
Glenn Jocher 73cf75faa8 Created using Colaboratory 2020-12-27 13:38:12 -08:00
Glenn Jocher b096221b1f Created using Colaboratory 2020-12-27 13:36:17 -08:00
Glenn Jocher c0ffcdf998 Display correct CUDA devices (#1776)
* Display correct CUDA devices

* cleanup
2020-12-24 13:01:35 -08:00
Glenn Jocher 8a81839cad Update tutorial.ipynb 2020-12-23 12:43:10 -08:00
Glenn Jocher 98a46d8e82 Update README.md (#1764) 2020-12-22 23:03:08 -08:00
Glenn Jocher 720afe68e9 FROM nvcr.io/nvidia/pytorch:20.12-py3 (#1763) 2020-12-22 20:53:44 -08:00
Glenn Jocher 9922c1183f leaf Variable inplace bug fix (#1759) 2020-12-22 17:27:40 -08:00
Glenn Jocher 3004fb5bc1 Automatic m.half() profile on x.half() 2020-12-21 15:20:33 -08:00
Glenn Jocher 0bd9c48609 Update torch_utils.py
FLOPS to GFLOPS
2020-12-21 13:29:52 -08:00
Glenn Jocher 394d1c89f3 Input channel yaml['ch'] addition (#1741) 2020-12-19 10:54:01 -08:00
Glenn Jocher 685d601308 Increase plot_labels() speed (#1736) 2020-12-18 18:05:38 -08:00
Glenn Jocher 49abc722fc Update profile_idetection() (#1727) 2020-12-18 01:19:17 -08:00
Glenn Jocher d5289b54c4 clean_str() function addition (#1674)
* clean_str() function addition

* cleanup

* add euro symbol €

* add closing exclamation (spanish)

* cleanup
2020-12-17 17:20:20 -08:00
Glenn Jocher 7e161d9774 Single class train update (#1719) 2020-12-17 12:02:03 -08:00
Glenn Jocher 6bd5e8bca7 nn.SiLU() export support (#1713) 2020-12-16 17:55:57 -08:00
Glenn Jocher c923fbff90 W&B artifacts feature addition (#1712)
* Log artifacts

* cleanup
2020-12-16 17:52:12 -08:00
Glenn Jocher 8bc0027afc Update loss criteria constructor (#1711) 2020-12-16 08:39:35 -08:00
Glenn Jocher 799724108f Update C3 module (#1705) 2020-12-15 22:13:08 -08:00
Glenn Jocher 7947c86b57 Update COCO train postprocessing (#1702) 2020-12-15 21:50:28 -08:00
Glenn Jocher 69ea70cd3b Add idetection_profile() function to plots.py (#1700) 2020-12-15 18:35:47 -08:00