Removed parameter Channels

It seems to be standarized, so it is now used as an additional check for
input signatures.
This commit is contained in:
raystlin
2025-04-16 08:19:58 +00:00
parent eca0bc5205
commit d082929dee
7 changed files with 37 additions and 33 deletions

View File

@@ -35,9 +35,8 @@ var modelsInfo = map[string]*tensorflow.ModelInfo{
},
"efficientnet-v2-tensorflow2-imagenet1k-m-classification-v2.tar.gz": &tensorflow.ModelInfo{
Input: &tensorflow.PhotoInput{
Height: 480,
Width: 480,
Channels: 3,
Height: 480,
Width: 480,
},
Output: &tensorflow.ModelOutput{
OutputsLogits: true,
@@ -50,9 +49,8 @@ var modelsInfo = map[string]*tensorflow.ModelInfo{
},
"inception-v3-tensorflow2-classification-v2.tar.gz": &tensorflow.ModelInfo{
Input: &tensorflow.PhotoInput{
Height: 299,
Width: 299,
Channels: 3,
Height: 299,
Width: 299,
},
Output: &tensorflow.ModelOutput{
OutputsLogits: true,