Commit Graph

338 Commits (d82e27fd174d1207eaabbe32c91c4f1d01cdd001)

Author SHA1 Message Date
cuicheng01 03a3da164b Update tnt.py 2021-07-01 03:33:27 +00:00
cuicheng01 f731ac545f fix initial bug of levit 2021-07-01 02:56:10 +00:00
cuicheng01 dfb8e26923
Merge pull request #990 from cuicheng01/develop
Update vision_transformer.py
2021-06-30 14:48:47 +08:00
cuicheng01 c492e1b2a2 Update vision_transformer.py 2021-06-29 13:08:40 +00:00
littletomatodonkey 4d496f787b fix class_dim 2021-06-29 10:12:40 +00:00
Walter 6fc9e5ad54
Merge pull request #979 from cuicheng01/develop
update new_user/ShuffleNetV2_x0_25.yaml
2021-06-29 11:05:23 +08:00
cuicheng01 8a31e4f4d7 update xception_deeplab.py pretrained url 2021-06-28 15:18:12 +00:00
Tingquan Gao 24727034d8
Update whl (#871)
* Update whl

* Optimize usage of CLI args
2021-06-22 11:12:26 +08:00
Tingquan Gao 27849b7a76
Update en doc of Hubserving (#900) 2021-06-22 11:08:55 +08:00
Aurelius84 d5c1700faf support to_static for benchmark 2021-06-21 06:14:06 +00:00
cuicheng01 8246563c5a Remove expired codes 2021-06-18 11:10:29 +00:00
cuicheng01 d5a4ad19e2 Remove 2021-06-17 02:52:15 +00:00
weishengyu e11cc5c8c1 remove useless comments 2021-06-16 22:13:32 +08:00
littletomatodonkey 35437e2236 fix distillation model infer and export model 2021-06-13 05:01:56 +00:00
cuicheng01 d8def84624 Update model_zoo 2021-06-11 13:58:50 +00:00
cuicheng01 f136ee1843
Merge branch 'develop_reg' into reg/add_twins_levit 2021-06-11 17:01:49 +08:00
jm12138 d28b8607ca update code format 2021-06-11 00:14:34 +08:00
jm12138 6055b717ca add code License 2021-06-11 00:11:20 +08:00
jm12138 bec1289fcd update __all__ export 2021-06-10 23:20:29 +08:00
jm12138 a950ec42b0 add the codes of TNT, HarDNet, RedNet and DLA models 2021-06-10 23:15:40 +08:00
gaotingquan 9e2a8e862a Add LeViT and Twins 2021-06-10 14:29:12 +00:00
littletomatodonkey edec759fa2 fix reader 2021-06-10 09:19:49 +00:00
littletomatodonkey 3b4f5f4dfc
add distillation and fix some apis (#810)
* fix save load and imagenet dataset
* refine trainer
2021-06-10 16:30:05 +08:00
Wei Shengyu a175ce5efe
Merge pull request #800 from cuicheng01/develop_reg
Add products configs
2021-06-08 10:40:03 +08:00
cuicheng01 05aa04f7d3 Add products configs 2021-06-07 11:48:18 +00:00
Felix eaf3b31577
Update circlemargin.py 2021-06-07 15:31:30 +08:00
weishengyu 25f6af99b2 fix format 2021-06-07 15:05:51 +08:00
weishengyu 296ba94c1c dbg 2021-06-07 15:05:26 +08:00
weishengyu 07089dcfe7 move identity_head to gears 2021-06-07 14:22:53 +08:00
weishengyu 91f72dcef7 dbg label 2021-06-05 17:56:40 +08:00
weishengyu ce43150ffe dbg 2021-06-05 17:45:31 +08:00
weishengyu 1c7a23c5e7 dbg 2021-06-05 13:44:58 +08:00
dongshuilong 8a8eb34d79 add vehicle neck and fix bugs 2021-06-04 16:44:24 +08:00
dongshuilong 505c9309a0 add vehicle neck 2021-06-04 14:54:34 +08:00
dongshuilong 76fcde59af add variant model example 2021-06-04 10:19:01 +08:00
weishengyu 582157f209 modify format and remove loss 2021-06-03 15:23:23 +08:00
weishengyu 586af751ac rename head -> gears 2021-06-03 15:19:39 +08:00
dongshuilong 435144b526 fix bugs imgnet and rec train 2021-06-03 12:24:48 +08:00
dongshuilong 452f532121 add CompCars train and fix bugs 2021-06-03 11:00:47 +08:00
dongshuilong bba0cf8fdb add CompCars train 2021-06-02 20:04:24 +08:00
cuicheng01 d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00:00
Bin Lu 6391c90b81
Delete ppcls/arch/neck directory 2021-06-01 11:35:44 +08:00
Bin Lu 85ca658ea2
Update __init__.py 2021-06-01 11:30:26 +08:00
Bin Lu 9c395ac802
Update __init__.py 2021-06-01 09:52:56 +08:00
Wei Shengyu 82ed94704d
Merge pull request #755 from Intsigstephon/develop_reg
add  head/neck and losses
2021-05-31 20:50:02 +08:00
Bin Lu f61a64aab6
Update arcmargin.py 2021-05-31 20:46:45 +08:00
Bin Lu 77a2c4571c
Update cosmargin.py 2021-05-31 20:46:16 +08:00
Bin Lu 6f2959af3a
Update arcmargin.py 2021-05-31 20:45:30 +08:00
weishengyu 546d63d58a dbg 2021-05-31 16:49:25 +08:00
weishengyu c749bdf93f replace reshape with Flatten layer in ResNet 2021-05-31 16:40:33 +08:00
dongshuilong 2fa808517d legendary models v0.1 2021-05-31 16:01:02 +08:00
Bin Lu d0edde7a94
Create circlemargin.py 2021-05-31 14:16:33 +08:00
Bin Lu 2177c3435c
Create cosmargin.py 2021-05-31 14:16:03 +08:00
Bin Lu 777f6600e0
Create fc.py 2021-05-31 14:15:37 +08:00
Bin Lu c0ea0f39bc
Create arcmargin.py 2021-05-31 14:15:14 +08:00
Bin Lu 5e064d0b24
Create __init__.py 2021-05-31 14:14:48 +08:00
Bin Lu dbb3e4bfc0
Create fc.py 2021-05-31 14:14:03 +08:00
Bin Lu 5148eabcca
Create __init__.py 2021-05-31 14:13:44 +08:00
Walter 4d246c20e4
Merge pull request #749 from Intsigstephon/develop_reg
add mobilenet_v1.py to legendary models
2021-05-31 11:22:42 +08:00
Bin Lu c2a7f190af
Delete ppcls/arch/head directory 2021-05-31 11:17:07 +08:00
Walter e469a975ea
Merge pull request #746 from FredHuang16/patch-1
add inception_v3.py
2021-05-31 11:09:27 +08:00
Walter 8ffd18b258
Merge pull request #748 from weisy11/develop_reg
modify hrnet
2021-05-31 11:08:51 +08:00
Bin Lu 119f4fcb34
Update fc.py 2021-05-28 19:28:53 +08:00
Bin Lu 3ebe1e506c
Create fc.py 2021-05-28 19:23:19 +08:00
Bin Lu f7f07e6ece
Create circlemargin.py 2021-05-28 19:22:58 +08:00
Bin Lu d3351aab96
Create cosmargin.py 2021-05-28 19:22:31 +08:00
Bin Lu e3d857d4c9
Create arcmargin.py 2021-05-28 19:21:24 +08:00
Bin Lu 693aa83fe2
Create __init__.py 2021-05-28 19:21:01 +08:00
Felix 79b21af761
Update inception_v3.py 2021-05-28 18:24:18 +08:00
weishengyu 8ede57a409 add MODEL_URLS 2021-05-28 16:38:45 +08:00
Felix 4ad209e5fe
Update inception_v3.py 2021-05-28 16:20:06 +08:00
weishengyu c05df5fa89 modify code style 2021-05-28 16:06:02 +08:00
Bin Lu 53bdc09baf
Update mobilenet_v1.py 2021-05-28 14:52:32 +08:00
Bin Lu 244e632d1a
Update mobilenet_v1.py 2021-05-28 14:51:07 +08:00
Bin Lu 2b84db06c1
Update mobilenet_v1.py 2021-05-28 14:50:21 +08:00
Bin Lu 084a27a15b
Update mobilenet_v1.py 2021-05-28 14:27:37 +08:00
Bin Lu 0e7ed899eb
Update mobilenet_v1.py 2021-05-28 14:21:01 +08:00
Bin Lu d3133089bd
Update mobilenet_v1.py 2021-05-28 14:04:03 +08:00
Bin Lu aa8cf9ba04
Update mobilenet_v1.py 2021-05-28 13:33:24 +08:00
Bin Lu f5ab94b4ed
Update mobilenet_v1.py 2021-05-28 13:30:12 +08:00
Bin Lu 5d155c8fab
Update mobilenet_v1.py
add pretrained and Model_urls
2021-05-28 13:28:11 +08:00
Walter decdb51bb0
Merge pull request #742 from cuicheng01/develop_reg
Add resnet.py in legendary_models
2021-05-28 11:36:24 +08:00
cuicheng01 0f3d321b81
Update resnet.py 2021-05-28 11:34:39 +08:00
cuicheng01 e6792d34da
Update resnet.py 2021-05-28 11:32:51 +08:00
cuicheng01 90321ce38d
Update resnet.py 2021-05-28 11:29:37 +08:00
cuicheng01 45b02b86af
Update resnet.py 2021-05-28 10:57:26 +08:00
cuicheng01 5671f9d9bb
Update resnet.py 2021-05-28 10:50:17 +08:00
Bin Lu 99adcce914
Update mobilenet_v1.py 2021-05-28 10:38:00 +08:00
Bin Lu 86eb2bc693
Update mobilenet_v1.py 2021-05-28 10:36:48 +08:00
Bin Lu 5f3456767d
Create mobilenet_v1.py 2021-05-27 19:13:03 +08:00
littletomatodonkey dd79f81fd7
[WIP]add arch init (#744)
* polish trainer
2021-05-27 18:41:44 +08:00
weishengyu dfe7f5b0ed dbg 2021-05-27 17:03:58 +08:00
weishengyu 014f1d1d27 add comment and change args of HRNet 2021-05-27 17:02:28 +08:00
Bin Lu f0e988e820
Update mobilenet_v1.py 2021-05-27 16:22:52 +08:00
Felix b8ac9ee31b
Update inception_v3.py 2021-05-27 14:54:30 +08:00
Felix 670c2bfb97
Update inception_v3.py 2021-05-27 14:42:45 +08:00
Bin Lu 06b83fd811
Update mobilenet_v1.py 2021-05-27 14:00:39 +08:00
Bin Lu 7b1dedb807
Update mobilenet_v1.py 2021-05-27 13:50:12 +08:00
Bin Lu 2d1a803964
Update mobilenet_v1.py 2021-05-27 13:36:33 +08:00
Bin Lu 8644331f07
add mobilenet_v1.py to legendary models 2021-05-27 12:10:37 +08:00
Felix d9eeb2d1bb
Update inception_v3.py
remove names and add nn.LayerList
2021-05-26 19:43:27 +08:00
weishengyu 3fa3e577ca remove names 2021-05-26 16:51:42 +08:00
weishengyu ad9caea2db dbg 2021-05-26 16:42:59 +08:00
weishengyu 7eda75225b modify code 2021-05-26 15:47:17 +08:00
Felix a62d5503fc
Create inception_v3.py
add inception_v3
2021-05-26 15:22:01 +08:00
weishengyu c7608cbf20 remove add_sublayers 2021-05-26 14:58:09 +08:00
weishengyu c9d5694cf2 dbg 2021-05-26 14:38:40 +08:00
weishengyu 1e938f7256 remove weight name and add_sublayer 2021-05-26 14:29:56 +08:00
littletomatodonkey 83056d44d5
add vgg network (#743)
* add vgg network

* fix base class

* fix relu and flatten
2021-05-26 11:00:54 +08:00
weishengyu 1d39358358 dbg 2021-05-26 00:31:08 +08:00
weishengyu 0e568006a8 dbg 2021-05-26 00:17:38 +08:00
weishengyu 64345cc0b1 dbg 2021-05-26 00:15:43 +08:00
weishengyu f645fac12d remove Layer1 2021-05-26 00:02:11 +08:00
weishengyu 15884fe78f remove layer1 2021-05-25 23:58:51 +08:00
weishengyu 94b1bae6ae dbg 2021-05-25 23:55:12 +08:00
weishengyu 1153ad6538 dbg 2021-05-25 23:54:23 +08:00
weishengyu 296294b91f dbg 2021-05-25 23:46:11 +08:00
weishengyu a0054caad4 dbg 2021-05-25 23:44:11 +08:00
weishengyu a2367a92b9 dbg 2021-05-25 23:43:41 +08:00
weishengyu 4d4557a32d dbg 2021-05-25 23:38:59 +08:00
weishengyu 9ab8639c3d remove transition 2021-05-25 23:37:43 +08:00
cuicheng01 465a8e5fd3
Create resnet.py 2021-05-25 19:18:14 +08:00
weishengyu b38b0f388f dbg 2021-05-25 17:23:17 +08:00
weishengyu be83d6ae6f use dict instead of list 2021-05-25 17:21:07 +08:00
weishengyu 4e5b60c082 remove name of ConvBN 2021-05-25 17:19:33 +08:00
weishengyu 9b61df62fd replace add_sublayer with nn.Sequential 2021-05-25 17:16:44 +08:00
weishengyu a16de6da0f remove name of conv and bn 2021-05-25 16:55:38 +08:00
weishengyu 2b150a2ffa dbg 2021-05-25 16:51:04 +08:00
weishengyu 88f92297fa dbg 2021-05-25 16:50:37 +08:00
weishengyu 79e9419fad dbg 2021-05-25 16:47:33 +08:00
weishengyu 611a40958a dbg 2021-05-25 16:40:20 +08:00
weishengyu 1499b0e585 add nn 2021-05-25 15:50:50 +08:00
weishengyu 6520a80176 add res_dict 2021-05-25 15:46:41 +08:00
weishengyu 1406b55aba add hrnet 2021-05-25 15:20:07 +08:00
weishengyu 010521e828 add folder 2021-05-25 14:45:50 +08:00
weishengyu 8f3df96eea move files 2021-05-25 14:39:53 +08:00
weishengyu 608b163d4f move stop_after out of update_sub, and rename update_sub to update_res 2021-05-24 19:49:06 +08:00
weishengyu f6da3bd60a move files 2021-05-24 11:42:24 +08:00