package/forensic
1
0
Fork 0
mirror of https://github.com/esimov/forensic.git synced 2025-09-26 20:41:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixing index out of range issue

Browse Source
This commit is contained in:
esimov
2019-03-13 07:28:54 +02:00
parent e4c68f10b7
commit 58c279656c
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
main.go
View File

@@ -104,6 +104,10 @@ func main() {
log.Fatal("Usage: forensic -in input.jpg -out out.jpg")
}
if *blockSize <= 1 {
log.Fatal("ERROR: the block size must be greater then 1.")
}
start := time.Now()
input, err := os.Open(*source)
@@ -172,7 +176,7 @@ func process(input image.Image, done chan struct{}) float64 {
}
bar := pb.StartNew(len(blocks))
bar.Prefix("Generate blocks: ")
bar.Prefix("Generate: ")
for _, block := range blocks {
// Average RGB value.
@@ -275,7 +279,7 @@ func process(input image.Image, done chan struct{}) float64 {
simBlocksNum := len(simBlocks)
forgedBlocksNum := len(forgedBlocks)
// precision indicated the detection accuracy.
// precision indicates the detection accuracy
var precision = 0.0
if forgedBlocksNum > 0 {
precision = 100 - (float64(forgedBlocksNum) / (float64(forgedBlocksNum + simBlocksNum)) * 100)