Commit Graph

987 Commits (8f6e447729e34a46fdbe9552fcfac705b82deac5)
 

Author SHA1 Message Date
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
Glenn Jocher 1bc0ef874a
Merge pull request #202 from ChristopherSTAN/master
Upload some scripts about VOC dataset
2020-07-04 16:13:15 -07:00
Glenn Jocher 72a1746938
Update get_voc.sh 2020-07-04 16:12:52 -07:00
Glenn Jocher 7df4fcebb1
Update voc.yaml 2020-07-04 16:12:17 -07:00
Glenn Jocher c0c5fd8121
Update voc.yaml 2020-07-04 16:11:14 -07:00
Glenn Jocher 9485f0b51f
Merge pull request #296 from NanoCode012/patch-1
Fix save error for multi-gpu in ema
2020-07-04 11:50:34 -07:00
NanoCode012 ccfa9373ae
Update to remove "module" attribute check 2020-07-04 23:12:57 +07:00
NanoCode012 f83521ab9a
Fix save error for multi-gpu 2020-07-04 16:50:33 +07:00