Commit Graph

282 Commits (5ba1de0cdcc414c69ceb7a4c45eb1e3895eca32a)
 

Author SHA1 Message Date
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
Glenn Jocher 5e2429e618 Merge remote-tracking branch 'origin/master' 2020-07-03 20:05:55 -07:00
Glenn Jocher 7bd7b2c090 update export.py 2020-07-03 20:05:50 -07:00
Glenn Jocher ce0c58f678 update compute_loss() 2020-07-03 18:56:07 -07:00
Glenn Jocher af41083d52 EMA FP16 fix #279 2020-07-03 16:57:08 -07:00
Glenn Jocher f767023c56 offset and balance update 2020-07-03 14:07:00 -07:00
Glenn Jocher 655895a838 module updates 2020-07-03 13:46:12 -07:00
Glenn Jocher df224a0d8f EMA bug fix #279 2020-07-03 11:56:14 -07:00
Glenn Jocher a9d20eba3e export.py update 2020-07-03 11:50:59 -07:00
Glenn Jocher 569fbc63c8 export.py update 2020-07-03 11:29:53 -07:00
Glenn Jocher 3bdea3f697 strip_optimizer() bug fix #253 2020-07-02 21:24:26 -07:00
Glenn Jocher 6ca3f35cd4 update .dockerignore 2020-07-02 16:41:03 -07:00
ChristopherSTAN d597513798
Create voc.yaml 2020-07-02 17:06:59 -04:00
ChristopherSTAN 8c3b829214
Add files via upload 2020-07-02 17:05:57 -04:00
ChristopherSTAN 4503d80449
Delete organize_VOC.py 2020-07-02 17:05:33 -04:00
ChristopherSTAN 0bd619b5aa
Delete voc.yml 2020-07-02 17:05:25 -04:00
ChristopherSTAN 389ed1f9d6
Delete VOC_split.py 2020-07-02 17:05:09 -04:00
ChristopherSTAN 2bcf3da848
Delete VOC2012.sh 2020-07-02 17:05:02 -04:00
ChristopherSTAN 570dcf05a9
Delete VOC2007.sh 2020-07-02 17:04:54 -04:00
Glenn Jocher e02a189a3a
Merge pull request #245 from yxNONG/patch-2
Unify the check point of single and multi GPU
2020-07-02 12:05:46 -07:00
Glenn Jocher f02481c73a
Update torch_utils.py 2020-07-02 12:03:45 -07:00
Glenn Jocher fc7c42723d
Update utils.py 2020-07-02 12:01:43 -07:00
Glenn Jocher 597ed4ce63
Update train.py 2020-07-02 12:00:55 -07:00
Glenn Jocher 9705c15da1
Update train.py 2020-07-02 11:37:48 -07:00
ChristopherSTAN a8d17ce712
Update train.py 2020-07-02 14:25:28 -04:00
Glenn Jocher 13f69777a6 typo fix 2020-07-02 09:26:03 -07:00
yxNONG 1aa2b67933
Update train.py 2020-07-02 13:51:52 +08:00
yxNONG 53cdaf6bf5
Update utils.py 2020-07-02 13:48:19 +08:00