Commit Graph

1037 Commits (803f51bceedb502e8f112b05911b805bf9ddac6b)
 

Author SHA1 Message Date
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
wanghaoyang0106 69c0d2da8a Merge branch 'master' of https://github.com/ultralytics/yolov5 2020-07-03 13:48:24 +08:00
wanghaoyang0106 b57f83d005 [bug fix] potential problem if img fed to model is in rectangular shape 2020-07-03 13:09:21 +08: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
Glenn Jocher a03adb5ab8 Merge remote-tracking branch 'origin/master' 2020-07-01 19:16:06 -07:00
Glenn Jocher 1fca7a7f24 autopad() update in common.py 2020-07-01 19:15:59 -07:00
Glenn Jocher 5323ad224d export.py update 2020-07-01 16:15:25 -07:00
Glenn Jocher a62a1c2c67 export.py update 2020-07-01 16:14:49 -07:00
Glenn Jocher f1d67f4110 update export.py 2020-07-01 15:46:15 -07:00
Glenn Jocher e74ccb2985 Merge remote-tracking branch 'origin/master' 2020-07-01 11:44:55 -07:00
Glenn Jocher b5659d1195 module updates 2020-07-01 11:44:49 -07:00
Glenn Jocher dfd63de20a
Merge pull request #255 from edurenye/ignore_torchscript
Add torchscript files to gitignore
2020-07-01 10:10:11 -07:00
edurenye 1c86c2cb66 Add torchscript files to gitignore 2020-07-01 10:04:26 +02:00
Glenn Jocher 86784cfdbf --resume bug fix #252 2020-06-30 21:43:53 -07:00
Glenn Jocher 0a08375a8a Merge remote-tracking branch 'origin/master' 2020-06-30 16:42:16 -07:00
Glenn Jocher 4e62eb90be check_img_size() int() update 2020-06-30 16:41:26 -07:00
Glenn Jocher 5c1c73ecda Merge remote-tracking branch 'origin/master' 2020-06-30 16:16:36 -07:00
Glenn Jocher ad4c22cbfe --resume bug fix #187 2020-06-30 16:16:29 -07:00
Glenn Jocher 2b7a11300e Merge remote-tracking branch 'origin/master' 2020-06-30 15:25:45 -07:00
Glenn Jocher a9dc0c2c29 update check_img_size() for model strides 2020-06-30 15:25:38 -07:00
Glenn Jocher b9da3ea46a Merge remote-tracking branch 'origin/master' 2020-06-30 14:08:14 -07:00
Glenn Jocher 3b16c865f0 assert --epochs 2020-06-30 14:08:08 -07:00
yxNONG cdb9bde181
Unify the check point of single and multi GPU
save the model.hyp etc to checkpoint when use multi GPU training
2020-06-30 19:06:28 +08:00
Glenn Jocher bfd51f62f8 Merge remote-tracking branch 'origin/master' 2020-06-29 17:10:41 -07:00
Glenn Jocher 9a9333d245 add replicate() to datasets.py 2020-06-29 17:10:33 -07:00