Added new parameters to model input.

New parameters have been added to define the input of the models:
* ResizeOperation: by default center-crop was being performed, now it is
  configurable.
* InputOrder: by default RGB was being used as the order for the array
  values of the input tensor, now it can be configured.
* InputInterval has been changed to InputIntervals (an slice). This
  means that every channel can have its own interval conversion.
* InputInterval can define now stddev and mean, because sometimes
  instead of adjusting the interval, the stddev and mean of the training
data should be use.
This commit is contained in:
raystlin
2025-07-15 13:31:31 +00:00
parent c682a94a07
commit adc4dc0f74
10 changed files with 800 additions and 40 deletions

View File

@@ -68,9 +68,11 @@ var modelsInfo = map[string]*tensorflow.ModelInfo{
},
"vision-transformer-tensorflow2-vit-b16-classification-v1.tar.gz": {
Input: &tensorflow.PhotoInput{
Interval: &tensorflow.Interval{
Start: -1.0,
End: 1.0,
Intervals: []tensorflow.Interval{
{
Start: -1.0,
End: 1.0,
},
},
},
Output: &tensorflow.ModelOutput{