refactor: inference result

This commit is contained in:
2021-12-31 18:47:21 +01:00
parent c5c87da714
commit ce58df8a6d
2 changed files with 12 additions and 10 deletions

View File

@ -7,7 +7,7 @@ import (
)
// LinearBin perform inverse linear_bin, taking
func LinearBin(arr []byte, n int, offset int, r float64) (float64, float64) {
func LinearBin(arr []int8, n int, offset int, r float64) (float64, float64) {
outputSize := len(arr)
type result struct {
score float64
@ -17,7 +17,7 @@ func LinearBin(arr []byte, n int, offset int, r float64) (float64, float64) {
var results []result
minScore := 0.2
for i := 0; i < outputSize; i++ {
score := float64(arr[i]) / 255.0
score := float64(int(arr[i])+128) / 255.0
if score < minScore {
continue
}