mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
modify tipc
This commit is contained in:
parent
d7e071fb8d
commit
add3ea238c
@ -28,23 +28,23 @@ from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_fro
|
|||||||
|
|
||||||
MODEL_URLS = {
|
MODEL_URLS = {
|
||||||
"SwinTransformerV2_tiny_patch4_window8_256":
|
"SwinTransformerV2_tiny_patch4_window8_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_tiny_patch4_window8_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_tiny_patch4_window8_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_tiny_patch4_window16_256":
|
"SwinTransformerV2_tiny_patch4_window16_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_tiny_patch4_window16_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_tiny_patch4_window16_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_small_patch4_window8_256":
|
"SwinTransformerV2_small_patch4_window8_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_small_patch4_window8_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_small_patch4_window8_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_small_patch4_window16_256":
|
"SwinTransformerV2_small_patch4_window16_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_small_patch4_window16_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_small_patch4_window16_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_base_patch4_window8_256":
|
"SwinTransformerV2_base_patch4_window8_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_base_patch4_window8_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_base_patch4_window8_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_base_patch4_window16_256":
|
"SwinTransformerV2_base_patch4_window16_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_base_patch4_window16_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_base_patch4_window16_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_base_patch4_window24_384":
|
"SwinTransformerV2_base_patch4_window24_384":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_base_patch4_window24_384_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_base_patch4_window24_384_pretrained.pdparams",
|
||||||
"SwinTransformerV2_large_patch4_window16_256":
|
"SwinTransformerV2_large_patch4_window16_256":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_large_patch4_window16_256_pretrained.pdparams",
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_large_patch4_window16_256_pretrained.pdparams",
|
||||||
"SwinTransformerV2_large_patch4_window24_384":
|
"SwinTransformerV2_large_patch4_window24_384":
|
||||||
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/SwinTransformerV2_large_patch4_window24_384_pretrained.pdparams"
|
"https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/SwinTransformerV2_large_patch4_window24_384_pretrained.pdparams"
|
||||||
}
|
}
|
||||||
|
|
||||||
__all__ = list(MODEL_URLS.keys())
|
__all__ = list(MODEL_URLS.keys())
|
||||||
@ -380,7 +380,7 @@ class SwinTransformerBlock(nn.Layer):
|
|||||||
assert L == H * W, "input feature has wrong size"
|
assert L == H * W, "input feature has wrong size"
|
||||||
|
|
||||||
shortcut = x
|
shortcut = x
|
||||||
x = x.reshape(shape=[B, H, W, C])
|
x = x.reshape([B, H, W, C])
|
||||||
|
|
||||||
# cyclic shift
|
# cyclic shift
|
||||||
if self.shift_size > 0:
|
if self.shift_size > 0:
|
||||||
@ -782,8 +782,8 @@ def _load_pretrained(pretrained,
|
|||||||
model,
|
model,
|
||||||
model_url,
|
model_url,
|
||||||
use_ssld=use_ssld,
|
use_ssld=use_ssld,
|
||||||
use_imagenet22k_pretrained=False,
|
use_imagenet22k_pretrained=use_imagenet22k_pretrained,
|
||||||
use_imagenet22kto1k_pretrained=False)
|
use_imagenet22kto1k_pretrained=use_imagenet22kto1k_pretrained)
|
||||||
elif isinstance(pretrained, str):
|
elif isinstance(pretrained, str):
|
||||||
load_dygraph_pretrain(model, pretrained, **kwargs)
|
load_dygraph_pretrain(model, pretrained, **kwargs)
|
||||||
else:
|
else:
|
||||||
@ -916,7 +916,7 @@ def SwinTransformerV2_base_patch4_window24_384(
|
|||||||
pretrained=False,
|
pretrained=False,
|
||||||
use_ssld=False,
|
use_ssld=False,
|
||||||
use_imagenet22k_pretrained=False,
|
use_imagenet22k_pretrained=False,
|
||||||
use_imagenet22kto1k_pretrained=False,
|
use_imagenet22kto1k_pretrained=True,
|
||||||
**kwargs):
|
**kwargs):
|
||||||
model = SwinTransformerV2(
|
model = SwinTransformerV2(
|
||||||
img_size=384,
|
img_size=384,
|
||||||
@ -941,7 +941,7 @@ def SwinTransformerV2_large_patch4_window16_256(
|
|||||||
pretrained=False,
|
pretrained=False,
|
||||||
use_ssld=False,
|
use_ssld=False,
|
||||||
use_imagenet22k_pretrained=False,
|
use_imagenet22k_pretrained=False,
|
||||||
use_imagenet22kto1k_pretrained=False,
|
use_imagenet22kto1k_pretrained=True,
|
||||||
**kwargs):
|
**kwargs):
|
||||||
model = SwinTransformerV2(
|
model = SwinTransformerV2(
|
||||||
img_size=256,
|
img_size=256,
|
||||||
@ -966,7 +966,7 @@ def SwinTransformerV2_large_patch4_window24_384(
|
|||||||
pretrained=False,
|
pretrained=False,
|
||||||
use_ssld=False,
|
use_ssld=False,
|
||||||
use_imagenet22k_pretrained=False,
|
use_imagenet22k_pretrained=False,
|
||||||
use_imagenet22kto1k_pretrained=False,
|
use_imagenet22kto1k_pretrained=True,
|
||||||
**kwargs):
|
**kwargs):
|
||||||
model = SwinTransformerV2(
|
model = SwinTransformerV2(
|
||||||
img_size=384,
|
img_size=384,
|
||||||
|
@ -123,7 +123,7 @@ DataLoader:
|
|||||||
order: ''
|
order: ''
|
||||||
sampler:
|
sampler:
|
||||||
name: DistributedBatchSampler
|
name: DistributedBatchSampler
|
||||||
batch_size: 32
|
batch_size: 128
|
||||||
drop_last: False
|
drop_last: False
|
||||||
shuffle: False
|
shuffle: False
|
||||||
loader:
|
loader:
|
||||||
|
@ -94,7 +94,7 @@ DataLoader:
|
|||||||
|
|
||||||
sampler:
|
sampler:
|
||||||
name: DistributedBatchSampler
|
name: DistributedBatchSampler
|
||||||
batch_size: 16
|
batch_size: 128
|
||||||
drop_last: False
|
drop_last: False
|
||||||
shuffle: True
|
shuffle: True
|
||||||
loader:
|
loader:
|
||||||
@ -123,7 +123,7 @@ DataLoader:
|
|||||||
order: ''
|
order: ''
|
||||||
sampler:
|
sampler:
|
||||||
name: DistributedBatchSampler
|
name: DistributedBatchSampler
|
||||||
batch_size: 16
|
batch_size: 128
|
||||||
drop_last: False
|
drop_last: False
|
||||||
shuffle: False
|
shuffle: False
|
||||||
loader:
|
loader:
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=384 -o PreProcess.transform_ops.1.CropImage.size=384
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=384 -o PreProcess.transform_ops.1.CropImage.size=384
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
@ -37,8 +37,8 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/S
|
|||||||
infer_model:../inference/
|
infer_model:../inference/
|
||||||
infer_export:True
|
infer_export:True
|
||||||
infer_quant:Fasle
|
infer_quant:Fasle
|
||||||
inference:python/predict_cls.py -c configs/inference_cls.yaml
|
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=292 -o PreProcess.transform_ops.1.CropImage.size=256
|
||||||
-o Global.use_gpu:True|False
|
-o Global.use_gpu:True
|
||||||
-o Global.enable_mkldnn:False
|
-o Global.enable_mkldnn:False
|
||||||
-o Global.cpu_num_threads:1
|
-o Global.cpu_num_threads:1
|
||||||
-o Global.batch_size:1
|
-o Global.batch_size:1
|
||||||
|
Loading…
x
Reference in New Issue
Block a user