public static MobileNet GetMobileNetV2(float multiplier, bool pretrained = false, Context ctx = null, string root = "", int classes = 1000, string prefix = "", ParameterDict @params = null) { var net = new MobileNet(multiplier, classes, prefix, @params); if (pretrained) { net.LoadParameters(ModelStore.GetModelFile($"mobilenetv2_{multiplier}"), ctx); } return(net); }
public static AlexNet GetAlexNet(bool pretrained = false, Context ctx = null, string root = "") { var net = new AlexNet(); if (ctx == null) { ctx = Context.CurrentContext; } if (pretrained) { net.LoadParameters(ModelStore.GetModelFile("alexnet", root), ctx); } return(net); }