mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
5,070 Commits
23 Branches
12 Tags
292 MiB
Commit Graph

413 Commits (df50d0a5a3a6021298902ea20512a54b62bb566f)

Author SHA1 Message Date
HydrogenSulfate dfd7749828 refine hard code 2022-04-21 14:37:00 +08:00
flytocc 7b020eb2aa add paper citation 2022-04-19 19:45:19 +08:00
flytocc c3018ebd39 add VAN-Classification code 2022-04-19 13:13:33 +08:00
weishengyu 0993147af7 dbg 2022-04-06 19:49:10 +08:00
cuicheng01 96fbbad038 update mobilevit.py 2022-04-01 09:06:46 +00:00
cuicheng01 44b9963b25 add mobilevit code 2022-04-01 08:51:42 +00:00
cuicheng01 4839915056 update cswin_transformer.py 2022-03-29 12:01:04 +00:00
cuicheng01 51e06d43f5 add cswin code and doc 2022-03-29 09:50:44 +00:00
Tingquan Gao 158c7b4e35 fix: fix bug that overrided eval() can't be called in distributed 
The eval() overrided can't be called because model is a DataParallel object in distributed.
 2022-03-04 16:52:57 +08:00
Wei Shengyu 1020f41899
use flatten instead of reshape for hrnet (#1709) 2022-02-28 19:12:53 +08:00
WangChen0902 7595ba6d70
add AFD (#1683) 
* add AFD
 2022-02-28 19:11:50 +08:00
gaotingquan 7abf42b7a1 fix: rm some repvgg models not supported 
remove RepVGG_B3, RepVGG_B2g2 and RepVGG_B3g2 that are not supported by PaddleClas.
 2022-02-25 23:50:29 +08:00
gaotingquan 4619d63273 fix: fix the download link, etc. 
fix the pretrained and inference download link;
add use_ssld arg to TNT model.
 2022-02-25 23:50:29 +08:00
gaotingquan 6963c25804 fix: fix bug 
bug is that error rasied by calling update_res() when sublayer does not have _save_sub_res_hook() because nn.Sequential and nn.LayerList is not TheseusLayer.
 2022-01-14 14:58:28 +08:00
gaotingquan a6a76f3d36 fix: args error 2022-01-14 14:58:28 +08:00
gaotingquan 43fce425ac fix: change the returned result from dict to list 2022-01-14 14:58:28 +08:00
gaotingquan 3b9f629220 feat: support specify return_stages to update_res() 2022-01-14 14:58:28 +08:00
gaotingquan 56c72a85a1 fix: fix lr 2022-01-07 11:52:30 +08:00
gaotingquan 025feb6de9 fix: fix -1 in dims 2022-01-07 11:52:30 +08:00
gaotingquan ee819393ac fix: fix -1 in dims and ignore swapdim in static 2022-01-07 11:52:30 +08:00
gaotingquan 00fb3f7519 feat: support PVTV2 2022-01-07 11:52:30 +08:00
gaotingquan a53b636829 feat: support PVTV2 2022-01-07 11:52:30 +08:00
littletomatodonkey aea712cc87
add dist of rec model (#1574) 
* add distillation loss func and rec distillation
 2022-01-05 19:25:36 +08:00
gaotingquan d6aeba6f56
fix: upgrade replace_sub() to upgrade_sublayer() 
supplement to PR #1581
 2021-12-30 10:59:20 +00:00
gaotingquan 71cb728b07 fix: rename to upgrade_sublayer() 2021-12-28 15:24:50 +08:00
gaotingquan 1e696ac22b fix: remove unnecessary register_hook() call & pre-commit 2021-12-28 15:24:50 +08:00
gaotingquan a86c4b290e add license 2021-12-28 15:24:50 +08:00
gaotingquan 721ac0bf61 refactor: simplify code 
1. remove WrapLayer and wrap_theseus;
2. support call update_res() one more;
3. optim parse_pattern_str() to return list of layer parsed.
 2021-12-28 15:24:50 +08:00
gaotingquan 56911b573b refactor: rename replace_sub() func to ? 2021-12-28 15:24:50 +08:00
gaotingquan 0f126b75da refactor: strengthen parse_pattern_str() func 2021-12-28 15:24:50 +08:00
gaotingquan 8d0b0d4b0a refactor: extract _parse_pattern_str() func 2021-12-28 15:24:50 +08:00
gaotingquan 18dec0744a fix: fix problems commented in reviewing 2021-12-28 15:24:50 +08:00
gaotingquan 41296972a4 fix: fix comments 2021-12-28 15:24:50 +08:00
gaotingquan f8ee6c0f86 fix: fix result returned by stop_after 2021-12-28 15:24:50 +08:00
gaotingquan cf205e1379 fix: fix result returned by _find_layers_handle 2021-12-28 15:24:50 +08:00
gaotingquan b0ae3a1210 fix: unify the pattern of layer's name 2021-12-28 15:24:50 +08:00
cuicheng01 897760f073
Merge pull request #1544 from cuicheng01/update_vit 
update vit
 2021-12-14 14:40:14 +08:00
cuicheng01 b3922c96f5
Merge pull request #1436 from TingquanGao/dev/update_whl 
update whl
 2021-12-14 12:15:46 +08:00
cuicheng01 83d1b411c5 update vit 2021-12-10 09:56:33 +00:00
weishengyu 6c5d1ebc28 add pruner and quanter for theseus 2021-12-09 14:51:40 +08:00
cuicheng01 5dad991c2a update se_resnext_vd.py 2021-11-30 15:06:05 +00:00
gaotingquan d53a06c625 fix: fix download link 2021-11-17 15:02:20 +08:00
gaotingquan ae6cb7d529 feat: support MixNet 2021-11-15 06:53:34 +00:00
cuicheng01 6e4bf593fb
Merge pull request #1390 from Intsigstephon/feature_binary_model 
add Binary general recog configure
 2021-11-12 15:25:00 +08:00
cuicheng01 fb631b38fb update esnet.py 2021-11-11 12:51:58 +00:00
cuicheng01 d868a63764 Add ESNet code and pretrained models 2021-11-11 06:00:32 +00:00
littletomatodonkey 67d982c30e
fix csp refer (#1404) 2021-11-05 13:54:10 +08:00
cuicheng01 7875e83d9e update vision_transformer.py 2021-11-05 02:53:10 +00:00
cuicheng01 08952deb88 update codes 2021-11-04 10:47:13 +00:00
Bin Lu 9f2f62f3ae
Update pp_lcnet_variant.py 2021-11-04 16:58:17 +08:00
Bin Lu 17fc4595b3
Create pp_lcnet_variant.py 2021-11-04 16:44:56 +08:00
Bin Lu cad5ce842a
Update __init__.py 2021-11-04 16:44:16 +08:00
stephon 6f13c5b798 rename PPLCNet_x2_5_Tanh 2021-11-04 08:37:21 +00:00
stephon df03b4a022 add Binary general reocg configure 2021-11-03 10:09:34 +00:00
gaotingquan 5025d09fd8 fix: fix a DeprecationWarning & style 2021-10-27 19:47:43 +08:00
Wei Shengyu 3c148cb56a
update return res method and fix amp bug (#1309) 
* fix amp bug; update return res method

* support index for list object for return res
 2021-10-15 18:25:50 +08:00
littletomatodonkey f8b420946c
fix cspnet (#1282) 
* fix cspnet

* minor fix
 2021-10-11 17:23:31 +08:00
cuicheng01 b7bab1e648 fix googlenet avg-pool 2021-09-23 08:29:07 +00:00
cuicheng01 48494ec051
Merge pull request #1213 from cuicheng01/develop 
Add PP-LCNet codes and configs
 2021-09-17 15:29:50 +08:00
Wei Shengyu d98b881603
Merge pull request #1208 from weisy11/fix_bug_of_theseus 
dbg theseus
 2021-09-15 14:04:51 +08:00
weishengyu e916259508 update return_res method 2021-09-15 11:03:46 +08:00
Walter 74622af482
Merge pull request #1093 from RainFrost1/slim 
添加slim功能
 2021-09-13 19:11:43 +08:00
cuicheng01 d470f34c64 Update code and docs 2021-09-13 03:18:37 +00:00
dongshuilong 8e85ef4775 update slim 2021-09-10 03:09:03 +00:00
cuicheng01 4150df912e Add LCNet codes and config 2021-09-07 16:14:40 +00:00
cuicheng01 3f0b406fb2 fix tnt inference bug when bs > 1 2021-09-07 03:46:29 +00:00
weishengyu 5131956d15 dbg theseus 2021-09-06 16:18:57 +08:00
dongshuilong 50f2547005 Merge branch 'develop' of https://github.com/PaddlePaddle/PaddleClas into slim 2021-09-03 03:19:14 +00:00
dongshuilong 68c1b089e8 update slim for new trainer 2021-08-27 09:32:37 +00:00
Bin Lu 35d2be15dc
Merge branch 'PaddlePaddle:develop' into develop 2021-08-25 13:53:14 +08:00
cuicheng01 8c8153e97f Update SwinTransformer pretrained url 2021-08-24 03:22:03 +00:00
Bin Lu c9912f0a04
Update vgg_variant.py 2021-08-24 10:53:13 +08:00
Bin Lu f878fb27e5
Update __init__.py 2021-08-23 19:19:19 +08:00
Bin Lu 26eda0efef
Update __init__.py 2021-08-23 19:08:06 +08:00
Bin Lu 0b9f0ad001
Create vgg_variant.py 2021-08-23 19:07:35 +08:00
Wei Shengyu 4ae4a4a903
Merge pull request #1116 from weisy11/develop 
update save sub layer result
 2021-08-18 14:12:43 +08:00
cuicheng01 2940e2a9b8
Merge pull request #1130 from cuicheng01/develop 
fix resnext101_wsl pretrain url bug
 2021-08-11 23:01:13 +08:00
cuicheng01 ac404c7a37
Update resnext101_wsl.py 2021-08-11 22:57:28 +08:00
Wei Shengyu 59be8e4674
Merge branch 'release/2.2' into develop 2021-08-09 11:07:31 +08:00
weishengyu 98f38fa0f7 dbg 2021-08-08 17:53:06 +08:00
weishengyu f41b09ef7f dbg 2021-08-08 17:28:57 +08:00
weishengyu 88a72b6735 dbg 2021-08-08 15:56:57 +08:00
weishengyu 2cb3bf66dd update wrap theseus rule 2021-08-08 15:55:34 +08:00
weishengyu 1985075550 dbg 2021-08-08 15:18:28 +08:00
weishengyu ef13f8c640 modify code 2021-08-08 15:04:26 +08:00
weishengyu 6dbbf8cc17 add theseus wrapper 2021-08-08 14:57:29 +08:00
weishengyu 4f8552aa4f dbg 2021-08-08 14:32:45 +08:00
weishengyu 8211880d10 dbg 2021-08-08 14:29:07 +08:00
weishengyu 132ec21fba reverse res_dict 2021-08-08 14:27:43 +08:00
weishengyu 56c513fdc7 remove disconnect 2021-08-08 13:29:30 +08:00
weishengyu 3b62e4d401 dbg 2021-08-08 13:14:30 +08:00
weishengyu 7e1aa2d358 dbg 2021-08-08 12:58:31 +08:00
weishengyu 6ebe7f09ed add return_inter flag 2021-08-08 01:52:18 +08:00
weishengyu 9790cc517f add return_dict to trainer 2021-08-08 01:49:26 +08:00
weishengyu bba9e0dfff add vgg theseus funcs 2021-08-08 01:07:49 +08:00
weishengyu 8c1515b5db add hook 2021-08-08 00:37:53 +08:00
weishengyu 0040a973eb add save res back 2021-08-07 23:00:32 +08:00
cuicheng01 3a9e6b3179 fix ResNeXt101_wsl bugs 2021-08-03 08:48:58 +00:00
cuicheng01 cc55d6371b fix ResNeXt101_wsl bugs 2021-08-03 08:46:25 +00:00
gaotingquan abb5d0fdf6 fix: fix @ -> matmul 2021-07-27 04:26:21 +00:00
gaotingquan 2f666cdfc9 fix: fix @ -> matmul 2021-07-27 04:10:10 +00:00
cuicheng01 95eb232a5a Add MixNet and ReXNet configs 2021-07-22 03:38:47 +00:00
cuicheng01 da1b58840e Update levit.py 2021-07-22 03:36:04 +00:00
cuicheng01 18592f5b88 Update gvt.py 2021-07-21 13:18:03 +00:00
cuicheng01 3a8715240c Update gvt.py 2021-07-21 13:16:23 +00:00
Tingquan Gao c4d131ebd3
Update gvt.py 2021-07-21 20:50:34 +08:00
Tingquan Gao e07674603e
Update gvt.py 2021-07-21 20:50:04 +08:00
cuicheng01 e6eeb09a82 Update Twins configs 2021-07-21 09:35:58 +00:00
cuicheng01 88d0d4ca9f Update Twins configs 2021-07-21 09:32:58 +00:00
littletomatodonkey 9d9cd3719e
add static training (#1037) 
* add static training

* fix typo

* add se fp16

* rm note

* fix loader

* fix cfg
 2021-07-15 10:30:07 +08:00
cuicheng01 3c9086c270 Update levit.py 2021-07-12 03:49:37 +00:00
cuicheng01 68d5bb0341 Add MixNet and ReXNet configs 2021-07-09 06:20:41 +00:00
cuicheng01 48ff7ec1f9
Merge branch 'PaddlePaddle:develop' into develop 2021-07-06 21:51:26 +08:00
cuicheng01 9db7262f0e fix LeViT_384 train bugs 2021-07-05 14:13:30 +00:00
dongshuilong 1b004933e1 fix resnet50_variant pretrained bugs 2021-07-05 12:16:48 +00:00
cuicheng01 f6c5625f25 Fix LeVit export_model bugs 2021-07-05 07:43:43 +00:00
cuicheng01 a6420f1cce Update tnt.py 2021-07-01 03:37:42 +00:00
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
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
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
dongshuilong 76fcde59af add variant model example 2021-06-04 10:19:01 +08:00
cuicheng01 d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00: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
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
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
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 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