mirror of
https://github.com/photoprism/photoprism.git
synced 2025-12-12 00:34:13 +01:00
5521d06bc0
This parameter allows us to rescale the input of the models because some of them need values between [0, 1] and other between [-1, 1].
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package tensorflow
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/wamuir/graft/tensorflow"
|
|
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
)
|
|
|
|
var defaultImageInput = &PhotoInput{
|
|
Height: 224,
|
|
Width: 224,
|
|
Channels: 3,
|
|
}
|
|
|
|
func TestConvertValue(t *testing.T) {
|
|
result := convertValue(uint32(98765432), &Interval{Start: -1, End: 1})
|
|
assert.Equal(t, float32(3024.8982), result)
|
|
}
|
|
|
|
func TestImageFromBytes(t *testing.T) {
|
|
var assetsPath = fs.Abs("../../../assets")
|
|
var examplesPath = assetsPath + "/examples"
|
|
|
|
t.Run("CatJpeg", func(t *testing.T) {
|
|
imageBuffer, err := os.ReadFile(examplesPath + "/cat_brown.jpg")
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
result, err := ImageFromBytes(imageBuffer, defaultImageInput)
|
|
assert.Equal(t, tensorflow.DataType(0x1), result.DataType())
|
|
assert.Equal(t, int64(1), result.Shape()[0])
|
|
assert.Equal(t, int64(224), result.Shape()[2])
|
|
})
|
|
t.Run("Document", func(t *testing.T) {
|
|
imageBuffer, err := os.ReadFile(examplesPath + "/Random.docx")
|
|
assert.Nil(t, err)
|
|
result, err := ImageFromBytes(imageBuffer, defaultImageInput)
|
|
|
|
assert.Empty(t, result)
|
|
assert.EqualError(t, err, "image: unknown format")
|
|
})
|
|
}
|