diff --git a/fluid/PaddleCV/video/models/attention_lstm/attention_lstm.py b/fluid/PaddleCV/video/models/attention_lstm/attention_lstm.py index 88bb6f334e8f32ff93e037b61fb7bcf673268f76..5d28dc47e297ba76e2b07fd69de2efa3ae6ccb0f 100755 --- a/fluid/PaddleCV/video/models/attention_lstm/attention_lstm.py +++ b/fluid/PaddleCV/video/models/attention_lstm/attention_lstm.py @@ -147,4 +147,5 @@ class AttentionLSTM(ModelBase): ] def weights_info(self): - return (None, None) + return ('attention_lstm_youtube8m', + 'https://paddlemodels.bj.bcebos.com/video_classification/attention_lstm_youtube8m.tar.gz') diff --git a/fluid/PaddleCV/video/models/stnet/stnet.py b/fluid/PaddleCV/video/models/stnet/stnet.py index e20ad0bd2adf21d6920a7660f86ff4a026b74060..c408aa0894265b2e7b8eccb8b49cae4c799018cc 100644 --- a/fluid/PaddleCV/video/models/stnet/stnet.py +++ b/fluid/PaddleCV/video/models/stnet/stnet.py @@ -128,6 +128,10 @@ class STNET(ModelBase): def pretrain_info(self): return ('ResNet50_pretrained', 'https://paddlemodels.bj.bcebos.com/video_classification/ResNet50_pretrained.tar.gz') + def weights_info(self): + return ('stnet_kinetics', + 'https://paddlemodels.bj.bcebos.com/video_classification/stnet_kinetics.tar.gz') + def load_pretrain_params(self, exe, pretrain, prog, place): def is_parameter(var): if isinstance(var, fluid.framework.Parameter): diff --git a/fluid/PaddleCV/video/models/tsn/tsn.py b/fluid/PaddleCV/video/models/tsn/tsn.py index 5bc8aba3886df138fc5111965b344d47325063cd..82fdb3279376a15e796df09818c343db24f048b4 100644 --- a/fluid/PaddleCV/video/models/tsn/tsn.py +++ b/fluid/PaddleCV/video/models/tsn/tsn.py @@ -132,6 +132,10 @@ class TSN(ModelBase): def pretrain_info(self): return ('ResNet50_pretrained', 'https://paddlemodels.bj.bcebos.com/video_classification/ResNet50_pretrained.tar.gz') + def weights_info(self): + return ('tsn_kinetics', + 'https://paddlemodels.bj.bcebos.com/video_classification/tsn_kinetics.tar.gz') + def load_pretrain_params(self, exe, pretrain, prog, place): def is_parameter(var): return isinstance(var, fluid.framework.Parameter) and (not ("fc_0" in var.name))