Store page metadata after capture
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user