Store page metadata after capture

This commit is contained in:
Pierre-Yves Nicolas
2026-01-09 15:24:15 +01:00
parent 0439971e57
commit fcdcea1891
11 changed files with 198 additions and 64 deletions

View File

@@ -14,11 +14,10 @@
*/
package org.fairscan.evaluation
import org.fairscan.imageprocessing.ExtractedDocument
import org.fairscan.imageprocessing.detectDocumentQuad
import org.fairscan.imageprocessing.extractDocument
import org.fairscan.imageprocessing.isColoredDocument
import org.fairscan.imageprocessing.scaledTo
import org.opencv.core.Mat
import org.opencv.imgcodecs.Imgcodecs
import java.io.File
@@ -61,11 +60,11 @@ object ColorDetectionEvaluator {
val quad = detectDocumentQuad(mask, isLiveAnalysis = false)
?.scaledTo(mask.width, mask.height, mat.width(), mat.height())
val document: Mat = if (quad != null) {
val extracted: ExtractedDocument = if (quad != null) {
extractDocument(mat, quad, 0, mask)
} else continue
val detected = isColoredDocument(mat, mask, quad)
val detected = extracted.pageAnalysis.isColored
nbProcessedImages++
@@ -73,7 +72,7 @@ object ColorDetectionEvaluator {
Imgcodecs.imwrite(inputOut.absolutePath, mat)
val outputOut = File(outputDir, "${imgName}_output.jpg")
Imgcodecs.imwrite(outputOut.absolutePath, document)
Imgcodecs.imwrite(outputOut.absolutePath, extracted.image)
results += ColorResult(
imgName,

View File

@@ -71,7 +71,7 @@ object DatasetEvaluator {
?.scaledTo(mask.width, mask.height, inputMat.width(), inputMat.height())
val corrected: Mat? = if (quad != null) {
extractDocument(inputMat, quad = quad, rotationDegrees = 0, mask)
extractDocument(inputMat, quad = quad, rotationDegrees = 0, mask).image
} else null
val inputOut = File(outputDir, "${e.name}_input.jpg")