test

Documentation

Index

• Constants
• func UnzipModels() (unzippedBaseDir string, unzippedModels []string, err error)

Constants

const (
	//MobileNetV2 testing model
	MobileNetV2 = "MobileNetV2.tf"
	//NASNetMobile testing model
	NASNetMobile = "NASNetMobile.tf"

	//MobileNetV2Input input operation name of MobileNetV2 model
	MobileNetV2Input = "serving_default_input_1"
	//NASNetMobileInput input operation name of NASNetMobile model
	NASNetMobileInput = "serving_default_input_2"
)

Functions

func UnzipModels

func UnzipModels() (unzippedBaseDir string, unzippedModels []string, err error)

UnzipModels unzips testing models into a temporary dir