Package examples.transferlearning

Functions

getFileFromResource
Link copied to clipboard
fun getFileFromResource(fileName: String): File
Content copied to clipboard

Converts resource string path to the file.

runImageRecognitionPrediction
Link copied to clipboard
fun runImageRecognitionPrediction(modelType: TFModels.CV<out GraphTrainableModel>, resizeTo: Pair<Int, Int> = Pair(224, 224))
Content copied to clipboard
runImageRecognitionTransferLearning
Link copied to clipboard
fun runImageRecognitionTransferLearning(modelType: TFModels.CV<out GraphTrainableModel>, epochs: Int = 2)
Content copied to clipboard
fun runImageRecognitionTransferLearningOnTopModel(modelType: TFModels.CV<out GraphTrainableModel>, epochs: Int = 2)
Content copied to clipboard