package/forensic
1
0
Fork 0
mirror of https://github.com/esimov/forensic.git synced 2025-10-04 16:12:39 +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") 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() start := time.Now()
input, err := os.Open(*source) input, err := os.Open(*source)
@@ -172,7 +176,7 @@ func process(input image.Image, done chan struct{}) float64 {
} }
bar := pb.StartNew(len(blocks)) bar := pb.StartNew(len(blocks))
bar.Prefix("Generate blocks: ") bar.Prefix("Generate: ")
for _, block := range blocks { for _, block := range blocks {
// Average RGB value. // Average RGB value.
@@ -275,7 +279,7 @@ func process(input image.Image, done chan struct{}) float64 {
simBlocksNum := len(simBlocks) simBlocksNum := len(simBlocks)
forgedBlocksNum := len(forgedBlocks) forgedBlocksNum := len(forgedBlocks)
// precision indicated the detection accuracy. // precision indicates the detection accuracy
var precision = 0.0 var precision = 0.0
if forgedBlocksNum > 0 { if forgedBlocksNum > 0 {
precision = 100 - (float64(forgedBlocksNum) / (float64(forgedBlocksNum + simBlocksNum)) * 100) precision = 100 - (float64(forgedBlocksNum) / (float64(forgedBlocksNum + simBlocksNum)) * 100)