Commit Graph
100 Commits
Author SHA1 Message Date
Glenn Jocher b9855f11dc update coco.yaml 2020-07-15 17:52:27 -07:00
Glenn Jocher 2dd43bc82a update Dockerfile 2020-07-15 17:40:19 -07:00
Glenn Jocher 1a7a168cec Merge remote-tracking branch 'origin/master' 2020-07-15 16:08:14 -07:00
Glenn Jocher 98f4a5af87 runs/exp increment bug fix #344 2020-07-15 16:07:40 -07:00
Glenn Jocher 0281c9cd18 Merge remote-tracking branch 'origin/master' 2020-07-15 10:39:38 -07:00
Glenn Jocher 03489aaafb comment tb_writer.add_hparams(hyp, {}) 2020-07-15 10:38:32 -07:00
Glenn Jocher a04050076e Update README.md 2020-07-15 00:27:30 -07:00
Glenn Jocher 95ef36bb74 Merge remote-tracking branch 'origin/master' 2020-07-15 00:18:56 -07:00
Glenn Jocher bcd727eb81 update README.md 2020-07-15 00:18:50 -07:00
Glenn Jocher f796708b50 Merge pull request #408 from seahawks8/patch-1
python3.8 numpy version
2020-07-14 16:55:16 -07:00
Glenn Jocher 611ec44359 Update README.md 2020-07-14 13:18:35 -07:00
Glenn Jocher 1e94bcf3d2 Merge pull request #384 from jancio/master
Log hyperparameters in tensorboard
2020-07-14 12:32:27 -07:00
Glenn Jocher 120d40c06a Update train.py
This updates the PR to a one-liner to minimize additions. Perhaps we can include opt in the future but let's start with this for now.
2020-07-14 12:32:08 -07:00
Glenn Jocher a1c8406af3 EMA and non_blocking=True 2020-07-13 20:19:10 -07:00
Glenn Jocher 2377e5f6ee FP16 EMA bug fix 2020-07-13 15:47:46 -07:00
Glenn Jocher 2b1892430f yolo.py bug fix 2020-07-13 15:21:19 -07:00
Glenn Jocher 02445d176d improved model.yaml source tracking 2020-07-13 14:35:47 -07:00
Glenn Jocher c80b249e67 update Dockerfile 2020-07-13 13:30:57 -07:00
Glenn Jocher 78c9420e09 --classes bug fix #387 2020-07-13 13:27:46 -07:00
Glenn Jocher 140d84cca1 comment updates 2020-07-13 12:17:52 -07:00
Glenn Jocher ea34f848a6 Merge remote-tracking branch 'origin/master' 2020-07-13 10:53:43 -07:00
Glenn Jocher 094079bd26 update detect.py --class argument #387 2020-07-13 10:53:34 -07:00
Glenn Jocher e169edfcf5 Update README.md 2020-07-12 14:28:19 -07:00
Glenn Jocher 8412d44dce update datasets.py to add detect.py --source glob_command feature 2020-07-12 14:22:15 -07:00
Glenn Jocher eae33303d6 update datasets.py LoadImages() path improvements and Mixup 2020-07-12 14:14:51 -07:00
Glenn Jocher 0afbb8d498 update test.py model.half() 2020-07-12 13:25:58 -07:00
Glenn Jocher 41ff83a98d update torch_utils.py to FP16 EMA 2020-07-12 12:18:00 -07:00
Glenn Jocher 8b38e6f429 update dataset comments 2020-07-12 10:37:06 -07:00
Glenn Jocher 1b1681bac9 Created using Colaboratory 2020-07-11 15:25:52 -07:00
Glenn Jocher 9d7c7784f2 Created using Colaboratory 2020-07-11 15:16:24 -07:00
Glenn Jocher a5818b280e expired pytorch hub cache bug fix #365 2020-07-11 15:04:53 -07:00
Glenn Jocher 8797246948 Merge remote-tracking branch 'origin/master' 2020-07-11 15:03:18 -07:00
Glenn Jocher 6e3c3b62b2 expired pytorch hub cache bug fix #365 2020-07-11 15:03:12 -07:00
Glenn Jocher 80d5775a57 Update README.md 2020-07-11 14:31:43 -07:00
Glenn Jocher f7a6d6c8d4 remove https:// prefix from curl commands fix #355 2020-07-11 13:00:24 -07:00
Glenn Jocher 01a73ec08e multi-gpu ckpt filesize bug fix #253 2020-07-11 12:39:27 -07:00
Glenn Jocher a586751904 multi-gpu ckpt filesize bug fix #253 2020-07-11 12:35:21 -07:00
Glenn Jocher 5de4e25d68 update tensorboard metric 2020-07-11 11:39:02 -07:00
Glenn Jocher 224e9a1d0b Merge remote-tracking branch 'origin/master' 2020-07-11 09:31:59 -07:00
Glenn Jocher 98fc483abc train.py results.txt to bucket bug fix 2020-07-11 09:31:53 -07:00
Glenn Jocher d6f6b33a49 update datasets.py for image size check during caching 2020-07-11 00:21:52 -07:00
Glenn Jocher 01a67a9742 Merge remote-tracking branch 'origin/master' 2020-07-10 20:36:26 -07:00
Glenn Jocher d3e786ed93 google_utils.py bug fix #355 2020-07-10 20:36:10 -07:00
Glenn Jocher 9006b85d21 update export.py torchscript file extension 2020-07-10 11:56:01 -07:00
Glenn Jocher 41bb70b325 update create_pretrained() 2020-07-10 00:25:35 -07:00
Glenn Jocher 1531795f5e avoid git status check in Docker bug fix #342 2020-07-10 00:11:50 -07:00
Glenn Jocher 1e9cf6a6e9 add file sizes to ckpt ops 2020-07-10 00:00:19 -07:00
Glenn Jocher c494a11128 Merge remote-tracking branch 'origin/master'
# Conflicts:
#	train.py
2020-07-09 22:09:40 -07:00
Glenn Jocher 1c13e67b33 evolution bug fix #346 2020-07-09 22:08:42 -07:00
Glenn Jocher 17b9d7cdda Merge pull request #344 from Laughing-q/master
update os.sep
2020-07-09 21:09:33 -07:00
Glenn Jocher 59beae98e8 add if isfile() to get_hash() 2020-07-09 20:57:24 -07:00
Glenn Jocher 6649770001 --rect test bug fix 2020-07-09 20:39:11 -07:00
Glenn Jocher d40e4781f1 --rect test bug patch 2020-07-09 20:36:30 -07:00
Glenn Jocher 520f5de6f0 Label caching foundational re-write #306 2020-07-09 20:07:16 -07:00
Glenn Jocher f310ca3bff Update issue templates 2020-07-09 17:36:26 -07:00
Glenn Jocher e16e9e43e1 new nc=len(names) check 2020-07-09 17:10:43 -07:00
Glenn Jocher cb527d3af9 new nc=len(names) check 2020-07-09 17:03:12 -07:00
Glenn Jocher dd8e742ece Merge pull request #334 from lorenzomammana/feature-multiple-datasets-training
Handle multiple datasets
2020-07-09 16:29:21 -07:00
Glenn Jocher dd33d2ab77 Update datasets.py 2020-07-09 16:28:20 -07:00
Glenn Jocher 2110f5ee89 update log_dir to runs/exp #107 2020-07-09 16:03:10 -07:00
Glenn Jocher 603ea0bfdc update log_dir to runs/exp #107 2020-07-09 15:58:07 -07:00
Glenn Jocher 72d5b58b9a disable LR plot to suppress warning message 2020-07-09 15:16:57 -07:00
Glenn Jocher 24c5a941f0 --resume EMA fix #292 2020-07-09 15:09:06 -07:00
Glenn Jocher 2b6209a9d5 Merge pull request #338 from alexstoken/hyp_save_bugfix
Move hyp and opt yaml save to top of train()
2020-07-09 14:35:32 -07: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
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
Glenn Jocher e8cf24b6c8 Initial model ensemble capability #318 2020-07-07 15:40:50 -07: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
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