Commit Graph

794 Commits

Author SHA1 Message Date
Alex Stoken 9d631408a2
Move hyp and opt yaml save to top of train()
Fixes bug where scaled values were saved in hyp.yaml, which would cause continuity issues with --resume
2020-07-09 16:18:55 -05:00
Glenn Jocher 13a09fa68a
Update datasets.py 2020-07-09 13:47:20 -07:00
Glenn Jocher 47bf1730fa
Update datasets.py 2020-07-09 13:45:55 -07:00
Glenn Jocher f9bab6b127
Update datasets.py 2020-07-09 13:36:23 -07:00
lorenzomammana 54a9e4f876 Refactor code to reduce duplication 2020-07-09 19:39:28 +02:00
lorenzomammana 94342acbe1 Handle multiple datasets 2020-07-09 11:52:12 +02:00
Glenn Jocher bf6f41567a hyperparameter printout update 2020-07-08 17:21:00 -07:00
Glenn Jocher 0fef3f6678
Merge pull request #104 from alexstoken/advanced_logging
Log command line options, hyperparameters, and weights per run in `runs/`
2020-07-08 17:01:32 -07:00
Glenn Jocher dc5e18390a
Merge branch 'master' into advanced_logging 2020-07-08 17:01:19 -07:00
Glenn Jocher abb024de04
Delete new_hyp.yaml 2020-07-08 16:59:06 -07:00
Glenn Jocher 6b134d93c5
Update train.py 2020-07-08 16:58:13 -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 cbe39a1dbb
Update utils.py 2020-07-08 16:14:32 -07:00
Glenn Jocher c654d18f00
Update utils.py 2020-07-08 16:11:22 -07:00
Glenn Jocher 16f6834486 update train.py and experimental.py 2020-07-08 14:23:34 -07:00
Glenn Jocher 1b9e28e72e
Update requirements.txt 2020-07-08 10:21:30 -07:00
Glenn Jocher 31b07b3550
Merge pull request #275 from wanghaoyang0106/master
[bug fix] potential problem if img fed to model is in rectangular shape
2020-07-08 00:07:54 -07:00
Glenn Jocher cd6609297c
Merge pull request #321 from dlawrences/master
Update CoreML model conversion to do pixel scaling as well
2020-07-08 00:04:46 -07:00
Laurentiu Diaconu 7c4d3cf51b
Merge pull request #1 from dlawrences/dev/coreml_convert_pxscaling
Updated coreml conversion to do pixel scaling between 0.0 and 1.0
2020-07-08 09:45:34 +03:00
Laurentiu Diaconu a9918fbf2b updated coreml conversion to do pixel scaling between 0.0 and 1.0 2020-07-08 09:43:33 +03:00
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07:00
Alex Stoken e6e7e7f1ac
Delete advanced_logging_test.ipynb
Remove from advanced_logging branch before merge.
2020-07-07 11:31:25 -05:00
Alex Stoken 43647514da Colab example/test of new features 2020-07-07 11:24:11 -05:00
Alex Stoken 95f0a56df7 Bug fix to get_latest_run when recent run is named 2020-07-07 11:19:49 -05:00
Alex Stoken 52bac22f09 Add in --resume functionality with option to specify path or to get most recent run 2020-07-07 10:42:28 -05:00
Alex Stoken 8b6dbb7cfc Add optimizer choice to hyp file 2020-07-07 10:30:48 -05: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 121d90b3f2 update fuse_conv_and_bn() 2020-07-06 11:46:10 -07: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 a40f615c6f .half() bug fix 2020-07-05 23:24:53 -07:00
Glenn Jocher 04bdbe4104 fuse update 2020-07-05 23:16:50 -07:00
Glenn Jocher 5ba1de0cdc update experimental.py with Ensemble() module 2020-07-05 15:02:56 -07:00
Glenn Jocher 38f5c1ad1d pruning and sparsity initial commit 2020-07-05 13:41:21 -07:00
Glenn Jocher 997ba7b346 import yaml in yolo.py 2020-07-05 12:50:04 -07:00
Glenn Jocher 8c43a6906d
Update README.md 2020-07-05 12:33:51 -07:00
Glenn Jocher 2541f77946 update detect.py 2020-07-05 11:57:48 -07:00
Glenn Jocher b7fe1d0622
Merge pull request #300 from Laughing-q/master
fix LR bug
2020-07-05 00:42:14 -07:00
Laughing 956511dafd
fix LR bug 2020-07-05 15:08:24 +08:00
Glenn Jocher 762b06de4b Merge remote-tracking branch 'origin/master' 2020-07-04 23:07:32 -07:00
Glenn Jocher 4e2d24602d update yolo.py 2020-07-04 23:07:26 -07:00
Glenn Jocher bb3c346916 model.yaml nc inherited from dataset.yaml 2020-07-04 17:51:54 -07:00
Glenn Jocher 1fdaa4987b update detect.py 2020-07-04 17:44:53 -07:00
Glenn Jocher 11ba5294dd update get_voc.sh 2020-07-04 17:36:53 -07:00
Glenn Jocher 659ad748c8 update get_voc.sh 2020-07-04 17:13:43 -07:00
Glenn Jocher fd0b1cc924 update voc.yaml 2020-07-04 16:50:11 -07:00
Glenn Jocher 45abf0c74f remove verbose tar unzip 2020-07-04 16:42:12 -07:00
Glenn Jocher 55e48603bf udpate dataset download instructions 2020-07-04 16:25:48 -07:00