mirror of
https://github.com/photoprism/photoprism.git
synced 2025-12-12 00:34:13 +01:00
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:
@@ -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{
|
||||
|
||||
Reference in New Issue
Block a user