refactor: convert input/output to uint8

This commit is contained in:
2021-12-31 19:36:35 +01:00
parent ce58df8a6d
commit ba9e54d1fb
2 changed files with 8 additions and 8 deletions

View File

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