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);
}
