def get_model():
if model_index == 0:
return mobilenet_v1.MobileNetV1()
elif model_index == 1:
return mobilenet_v2.MobileNetV2()
elif model_index == 2:
return mobilenet_v3_large.MobileNetV3Large()
elif model_index == 3:
return mobilenet_v3_small.MobileNetV3Small()
elif model_index == 4:
return efficientnet.efficient_net_b0()
elif model_index == 5:
return efficientnet.efficient_net_b1()
elif model_index == 6:
return efficientnet.efficient_net_b2()
elif model_index == 7:
return efficientnet.efficient_net_b3()
elif model_index == 8:
return efficientnet.efficient_net_b4()
elif model_index == 9:
return efficientnet.efficient_net_b5()
elif model_index == 10:
return efficientnet.efficient_net_b6()
elif model_index == 11:
return efficientnet.efficient_net_b7()
elif model_index == 12:
return resnext.ResNeXt50()
elif model_index == 13:
return resnext.ResNeXt101()
elif model_index == 14:
return inception_v4.InceptionV4()
elif model_index == 15:
return inception_resnet_v1.InceptionResNetV1()
elif model_index == 16:
return inception_resnet_v2.InceptionResNetV2()
def get_model():
if model_index == 0:
return mobilenet_v1.MobileNetV1()
elif model_index == 1:
return mobilenet_v2.MobileNetV2()
elif model_index == 2:
return mobilenet_v3_large.MobileNetV3Large()
elif model_index == 3:
return mobilenet_v3_small.MobileNetV3Small()
elif model_index == 4:
return efficientnet.efficient_net_b0()
elif model_index == 5:
return efficientnet.efficient_net_b1()
elif model_index == 6:
return efficientnet.efficient_net_b2()
elif model_index == 7:
return efficientnet.efficient_net_b3()
elif model_index == 8:
return efficientnet.efficient_net_b4()
elif model_index == 9:
return efficientnet.efficient_net_b5()
elif model_index == 10:
return efficientnet.efficient_net_b6()
elif model_index == 11:
return efficientnet.efficient_net_b7()
elif model_index == 12:
return resnext.ResNeXt50()
elif model_index == 13:
return resnext.ResNeXt101()
elif model_index == 14:
return inception_v4.InceptionV4()
elif model_index == 15:
return inception_resnet_v1.InceptionResNetV1()
elif model_index == 16:
return inception_resnet_v2.InceptionResNetV2()
elif model_index == 17:
return se_resnet.se_resnet_50()
elif model_index == 18:
return se_resnet.se_resnet_101()
elif model_index == 19:
return se_resnet.se_resnet_152()
elif model_index == 20:
return squeezenet.SqueezeNet()
elif model_index == 21:
return densenet.densenet_121()
elif model_index == 22:
return densenet.densenet_169()
elif model_index == 23:
return densenet.densenet_201()
elif model_index == 24:
return densenet.densenet_264()
elif model_index == 25:
return shufflenet_v2.shufflenet_0_5x()
elif model_index == 26:
return shufflenet_v2.shufflenet_1_0x()
elif model_index == 27:
return shufflenet_v2.shufflenet_1_5x()
elif model_index == 28:
return shufflenet_v2.shufflenet_2_0x()
def export_model(args):
if args.data == "mnist":
import paddle.dataset.mnist as reader
train_reader = reader.train()
val_reader = reader.test()
class_dim = 10
image_shape = "1,28,28"
elif args.data == "imagenet":
import imagenet_reader as reader
train_reader = reader.train()
val_reader = reader.val()
class_dim = 1000
image_shape = "3,224,224"
elif args.data == "fruit_veg":
import reader_cv2 as reader
train_reader = reader.train(settings=args)
val_reader = reader.val(settings=args)
class_dim = 23
image_shape = "3,224,224"
resize_short_size = 256
elif args.data == "yolov3-384":
import reader_cv2 as reader
train_reader = reader.train(settings=args)
val_reader = reader.val(settings=args)
class_dim = 80
image_shape = "3,384,384"
else:
raise ValueError("{} is not supported.".format(args.data))
image_shape = [int(m) for m in image_shape.split(",")]
image = fluid.data(name='image',
shape=[None] + image_shape,
dtype='float32')
assert args.model in model_list, "{} is not in lists: {}".format(
args.model, model_list)
# model definition
# model = models.__dict__[args.model]()
model = Model.InceptionV4()
out = model.net(input=image, class_dim=class_dim)
val_program = fluid.default_main_program().clone(for_test=True)
place = fluid.CUDAPlace(0) if args.use_gpu else fluid.CPUPlace()
exe = fluid.Executor(place)
exe.run(fluid.default_startup_program())
if args.pretrained_model:
def if_exist(var):
return os.path.exists(os.path.join(args.pretrained_model,
var.name))
fluid.io.load_vars(exe, args.pretrained_model, predicate=if_exist)
else:
assert False, "args.pretrained_model must set"
fluid.io.save_inference_model('./inference_model/' + args.model,
feeded_var_names=[image.name],
target_vars=[out],
executor=exe,
main_program=val_program,
model_filename='model',
params_filename='weights')
def get_model():
if model_index == 0:
return mobilenet_v1.MobileNetV1()
elif model_index == 1:
return mobilenet_v2.MobileNetV2()
elif model_index == 2:
return mobilenet_v3_large.MobileNetV3Large()
elif model_index == 3:
return mobilenet_v3_small.MobileNetV3Small()
elif model_index == 4:
return efficientnet.efficient_net_b0()
elif model_index == 5:
return efficientnet.efficient_net_b1()
elif model_index == 6:
return efficientnet.efficient_net_b2()
elif model_index == 7:
return efficientnet.efficient_net_b3()
elif model_index == 8:
return efficientnet.efficient_net_b4()
elif model_index == 9:
return efficientnet.efficient_net_b5()
elif model_index == 10:
return efficientnet.efficient_net_b6()
elif model_index == 11:
return efficientnet.efficient_net_b7()
elif model_index == 12:
return resnext.ResNeXt50()
elif model_index == 13:
return resnext.ResNeXt101()
elif model_index == 14:
return inception_v4.InceptionV4()
elif model_index == 15:
return inception_resnet_v1.InceptionResNetV1()
elif model_index == 16:
return inception_resnet_v2.InceptionResNetV2()
elif model_index == 17:
return se_resnet.se_resnet_50()
elif model_index == 18:
return se_resnet.se_resnet_101()
elif model_index == 19:
return se_resnet.se_resnet_152()
elif model_index == 20:
return squeezenet.SqueezeNet()
elif model_index == 21:
return densenet.densenet_121()
elif model_index == 22:
return densenet.densenet_169()
elif model_index == 23:
return densenet.densenet_201()
elif model_index == 24:
return densenet.densenet_264()
elif model_index == 25:
return shufflenet_v2.shufflenet_0_5x()
elif model_index == 26:
return shufflenet_v2.shufflenet_1_0x()
elif model_index == 27:
return shufflenet_v2.shufflenet_1_5x()
elif model_index == 28:
return shufflenet_v2.shufflenet_2_0x()
elif model_index == 29:
return resnet.resnet_18()
elif model_index == 30:
return resnet.resnet_34()
elif model_index == 31:
return resnet.resnet_50()
elif model_index == 32:
return resnet.resnet_101()
elif model_index == 33:
return resnet.resnet_152()
elif model_index == 34:
return vgg16.VGG16()
elif model_index == 35:
return vgg16_mini.VGG16()
elif model_index == 36:
return VGG16_self.VGG16()
elif model_index == 10086:
return diy_resnet.resnet_50()
else:
raise ValueError("The model_index does not exist.")
