Commit Graph

63 Commits (a12698f4c952d8804e7d0a5a70940202ff4844af)

Author SHA1 Message Date
Glenn Jocher c171e45cda Update test.py with set() (#806) 2020-08-21 15:03:03 -07:00
Glenn Jocher da35727c68
--save-txt bug fix (#763) 2020-08-17 17:14:07 -07:00
Glenn Jocher f5da528d28 reformat code 2020-08-14 11:53:44 -07:00
Glenn Jocher 916d4aad9a
v3.0 Release (#725)
* initial commit

* remove yolov3-spp from test.py study

* update study --img range

* update mAP

* cleanup and speed updates

* update README plot
2020-08-13 14:25:05 -07:00
NanoCode012 0892c44bc4
Fix Logging (#719)
* Add logging setup

* Fix fusing layers message

* Fix logging does not have end

* Add logging

* Change logging to use logger

* Update yolo.py

I tried this in a cloned branch, and everything seems to work fine

* Update yolo.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -07:00
Glenn Jocher 41523e2c91
Dataset autodownload feature addition (#685)
* initial commit

* move download scripts into data/scripts

* new check_dataset() function in general.py

* move check_dataset() out of with context

* Update general.py

* DDP update

* Update general.py
2020-08-09 20:52:57 -07:00
Jirka Borovec d5b6416c87
Explicit Imports (#498)
* expand imports

* optimize

* miss

* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher 43a616a955 PyTorch 1.6.0 compatability updates 2020-07-28 15:38:56 -07:00
Glenn Jocher 07a82f4d44
Update ci-testing.yml (#445)
* Update ci-testing.yml

* Update ci-testing.yml

* Update requirements.txt

* Update requirements.txt

* Update google_utils.py

* Update test.py

* Update ci-testing.yml
2020-07-19 15:44:11 -07:00
Glenn Jocher b40852d5a5 update test.py 2020-07-18 11:31:22 -07:00
Glenn Jocher 37acbdc0b6 update test.py --save-txt 2020-07-15 20:03:41 -07:00
Glenn Jocher b8c2da4a0d update test.py --save-txt 2020-07-15 20:00:48 -07:00
Glenn Jocher 1ce686edfc update test.py 2020-07-15 19:06:39 -07:00
Glenn Jocher 6bd9218922 update test.py 2020-07-15 19:04:15 -07:00
Glenn Jocher aa08b2bd6f test.py --task test updates 2020-07-15 18:25:34 -07:00
Glenn Jocher a1c8406af3 EMA and non_blocking=True 2020-07-13 20:19:10 -07:00
Glenn Jocher 0afbb8d498 update test.py model.half() 2020-07-12 13:25:58 -07:00
Glenn Jocher bf6f41567a hyperparameter printout update 2020-07-08 17:21:00 -07:00
Glenn Jocher dc5e18390a
Merge branch 'master' into advanced_logging 2020-07-08 17:01:19 -07:00
Glenn Jocher 22ab1c295f
Update test.py 2020-07-08 16:29:31 -07:00
Glenn Jocher c3d3e6b776
Update test.py 2020-07-08 16:27:52 -07:00
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07:00
Alex Stoken 5ac517b22a Remove duplicate `verbose` arg in test.py 2020-07-07 10:09:53 -05:00
Alex Stoken f517ba81c7
Merge branch 'master' into advanced_logging 2020-07-06 16:52:11 -05:00
Glenn Jocher 6b95d6d2c0 .to(device) bug fix 2020-07-05 23:37:36 -07:00
Glenn Jocher a62333e808 test.py .fuse() update 2020-07-05 23:27:03 -07:00
Glenn Jocher 04bdbe4104 fuse update 2020-07-05 23:16:50 -07:00
Glenn Jocher a9dc0c2c29 update check_img_size() for model strides 2020-06-30 15:25:38 -07:00
Alex Stoken e18e6811dc
Merge branch 'master' into advanced_logging 2020-06-27 10:13:03 -05:00
Glenn Jocher 22fb2b0c25 refactor dataloader 2020-06-26 18:56:13 -07:00
Glenn Jocher e670a3353b update test.py for FP16 testing during training 2020-06-25 15:09:04 -07:00
Glenn Jocher 227aa735dc updates 2020-06-24 13:20:14 -07:00
Glenn Jocher b8557f87e3 add stride to datasets.py 2020-06-24 13:02:27 -07:00
Glenn Jocher a63e1c9b3c study update 2020-06-22 23:23:50 -07:00
Glenn Jocher db898131f9 study update 2020-06-22 23:07:22 -07:00
Glenn Jocher 6c1b87a42e update google_utils import 2020-06-22 23:00:23 -07:00
Glenn Jocher 1f1917ef56 remove fast, add merge 2020-06-21 13:37:11 -07:00
Alex Stoken 9b7386f603 Add save_dir arg to test.test, use arg as location for saving batch jpgs 2020-06-17 16:08:46 -05:00
Glenn Jocher 2daa412575 train with multi-gpu half test bug fix #99 2020-06-16 16:28:25 -07:00
Glenn Jocher d1e5716610 train with multi-gpu half test bug fix #99 2020-06-16 16:22:58 -07:00
Glenn Jocher 72d06147b2 Multi-GPU disabled in test.py 2020-06-16 14:42:30 -07:00
Glenn Jocher 7c6fd4d670 --verbose bug fix #84 2020-06-16 10:22:22 -07:00
Glenn Jocher db2c3acd3a updated testing settings, rebalanced towards FP16 latency 2020-06-15 16:55:57 -07:00
Glenn Jocher 0b514daced FP16 test loss bug fix 2020-06-15 15:26:29 -07:00
Glenn Jocher a1748a8d6e test during training default to FP16 2020-06-15 15:20:27 -07:00
Glenn Jocher 76ca367a01 FP16 to FP32 ckpt load 2020-06-15 13:32:45 -07:00
Glenn Jocher c5966abba8 glob search bug fix #77 2020-06-15 12:08:57 -07:00
Glenn Jocher e7c24e04ac half() fix #66 2020-06-15 00:43:57 -07:00
Glenn Jocher 111994956e switch default inference to FP16 on GPU 2020-06-14 19:25:39 -07:00
Glenn Jocher 157aff2854 FP16 inference fix 2020-06-14 18:45:20 -07:00