提交 ca2aeeee 编写于 作者: G gaotingquan

Revert "Fix a bug about inference"

This reverts commit 17c66f42.

The bug had been fixed by another way.
上级 69f602e2
...@@ -28,7 +28,6 @@ import paddle ...@@ -28,7 +28,6 @@ import paddle
from paddle.distributed import ParallelEnv from paddle.distributed import ParallelEnv
import paddle.nn.functional as F import paddle.nn.functional as F
def parse_args(): def parse_args():
def str2bool(v): def str2bool(v):
return v.lower() in ("true", "t", "1") return v.lower() in ("true", "t", "1")
...@@ -102,11 +101,8 @@ def main(): ...@@ -102,11 +101,8 @@ def main():
place = paddle.CPUPlace() place = paddle.CPUPlace()
paddle.disable_static(place) paddle.disable_static(place)
if "EfficientNet" in args.model:
net = architectures.__dict__[args.model](is_test=True)
else:
net = architectures.__dict__[args.model]()
net = architectures.__dict__[args.model]()
load_dygraph_pretrain(net, args.pretrained_model, args.load_static_weights) load_dygraph_pretrain(net, args.pretrained_model, args.load_static_weights)
image_list = get_image_list(args.image_file) image_list = get_image_list(args.image_file)
for idx, filename in enumerate(image_list): for idx, filename in enumerate(image_list):
......
...@@ -61,6 +61,7 @@ def create_model(architecture, classes_num): ...@@ -61,6 +61,7 @@ def create_model(architecture, classes_num):
Args: Args:
architecture(dict): architecture information, architecture(dict): architecture information,
name(such as ResNet50) is needed name(such as ResNet50) is needed
image(variable): model input variable
classes_num(int): num of classes classes_num(int): num of classes
Returns: Returns:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册