apps/lpms
1
0
Fork 0
mirror of https://github.com/livepeer/lpms synced 2025-09-26 19:51:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

add benchmarking code for mpeg-7

Browse Source
This commit is contained in:
gpu
2022-09-01 11:59:54 +08:00
parent f21430428b
commit 688d5aaacb
2 changed files with 45 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cmd/example/analrenamer/analrenamer.go
View File

@@ -16,7 +16,7 @@ func selectFile(files *[]string) filepath.WalkFunc {
if err != nil {
return err
}
if info.IsDir() || (filepath.Ext(path) != ".ts" /*&& filepath.Ext(path) != ".hash"*/) {
if info.IsDir() || (filepath.Ext(path) != ".hash" /*&& filepath.Ext(path) != ".hash"*/) {
return nil
}
*files = append(*files, path)
@@ -49,9 +49,10 @@ func main() {
var vinfo ffmpeg.VideoInfo
vinfo, _ = ffmpeg.GetVideoInfoByPath(infiles[i])
if vinfo.Audiosum[0] == 0 && vinfo.Audiosum[1] == 0 && vinfo.Audiosum[2] == 0 && vinfo.Audiosum[3] == 0 {
fmt.Printf("%v", vinfo)
/*if vinfo.Audiosum[0] == 0 && vinfo.Audiosum[1] == 0 && vinfo.Audiosum[2] == 0 && vinfo.Audiosum[3] == 0 {
missaudio++
}
}*/
sl := strings.Split(infiles[i], "-")

43
cmd/example/analsigncomp/analsigncomp.go
View File

@@ -4,6 +4,7 @@ import (
"encoding/csv"
"flag"
"fmt"
"math/rand"
"os"
"path/filepath"
"sort"
@@ -32,6 +33,10 @@ func main() {
outcsv := "compresult.csv"
//indir := os.Args[0]
indir := "/home/gpu/tvideo/fastverifyfaildata/"
//equal, _ := ffmpeg.CompareSignatureByPath(indir+"1008137096162971958-ewr-trustphase1.hash", indir+"1001979456868321564-ewr-trustphase1.hash")
//fmt.Println("equal %v\n", equal)
if indir == "" {
panic("Usage: <input directory>")
}
@@ -53,13 +58,14 @@ func main() {
csvrecorder := csv.NewWriter(fwriter)
defer csvrecorder.Flush()
//write header
columnheader := []string{"filepath1", "filepath2", "equal"}
columnheader := []string{"filepath1", "filepath2", "predict_lab", "true_lab"}
_ = csvrecorder.Write(columnheader)
okaycount := 0
failcount := 0
for i := 0; i < len(infiles)/2; i++ {
//for i := 0; i < len(infiles)/2; i++ {
for i := 0; i < 5000; i++ {
var linestr []string
_, filename1 := filepath.Split(infiles[i*2])
@@ -75,6 +81,39 @@ func main() {
linestr = append(linestr, "0")
failcount++
}
//true label
linestr = append(linestr, "1")
csvrecorder.Write(linestr)
}
for i := 0; i < 5000; i++ {
var linestr []string
_, filename1 := filepath.Split(infiles[i*2])
randid := rand.Intn(i*2 + 100)
if randid == i*2 || randid == (i*2+1) || randid >= 5000 {
i--
continue
}
_, filename2 := filepath.Split(infiles[randid])
linestr = append(linestr, filename1)
linestr = append(linestr, filename2)
fmt.Printf("%v vs %v \n", filename1, filename2)
equal, _ := ffmpeg.CompareSignatureByPath(infiles[i*2], infiles[randid])
if equal {
linestr = append(linestr, "1")
failcount++
} else {
linestr = append(linestr, "0")
okaycount++
}
//true label
linestr = append(linestr, "0")
csvrecorder.Write(linestr)
}