dnutiu/ImageTagger
dnutiu/ImageTagger
1
1
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

add log when model or categories file is not found

Browse source
This commit is contained in:
Denis-Cosmin NUTIU 2024-03-23 21:36:55 +02:00
parent 0832a86810
commit 6bda1dbe06
1 changed files with 20 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/main/kotlin/com/nuculabs/dev/imagetagger/tag_prediction/ImageTagsPrediction.kt
View file

@ -19,17 +19,28 @@ class ImageTagsPrediction private constructor() {
private var modelClasses: MutableList<String> = mutableListOf()
init {
logger.info("Loaded ML model.")
ImageTagsPrediction::class.java.getResourceAsStream("/AIModels/prediction.onnx").let { modelFile ->
ortSession = ortEnv.createSession(
modelFile!!.readBytes(),
OrtSession.SessionOptions()
try {
logger.info("Loaded ML model.")
ImageTagsPrediction::class.java.getResourceAsStream("/AIModels/prediction.onnx").let { modelFile ->
ortSession = ortEnv.createSession(
modelFile!!.readBytes(),
OrtSession.SessionOptions()
)
}
ImageTagsPrediction::class.java.getResourceAsStream("/AIModels/prediction_categories.txt")
.let { classesFile ->
modelClasses.addAll(0, classesFile!!.bufferedReader().readLines())
}
logger.info("Loaded ${modelClasses.size} model classes.")
} catch (e: NullPointerException) {
logger.severe(
"Failed to load model file or categories file. If you're building the project from " +
"source, please follow the instructions from the README.md: " +
"https://github.com/dnutiu/ImageTagger." +
"Exception ${e.message}"
)
throw e
}
ImageTagsPrediction::class.java.getResourceAsStream("/AIModels/prediction_categories.txt").let { classesFile ->
modelClasses.addAll(0, classesFile!!.bufferedReader().readLines())
}
logger.info("Loaded ${modelClasses.size} model classes.")
}
private fun processImage(bufferedImage: BufferedImage): Array<Array<Array<FloatArray>>> {