Closed steve21124 closed 7 years ago
Yolo has grid method but InceptionV3 does not. MLMultiArray ` class TinyYOLOOutput : MLFeatureProvider { let grid: MLMultiArray
var featureNames: Set<String> { get { return ["grid"] } } func featureValue(for featureName: String) -> MLFeatureValue? { if (featureName == "grid") { return MLFeatureValue(multiArray: grid) } return nil } init(grid: MLMultiArray) { self.grid = grid }
} `
You might be able to train Inception-v3 the same way as YOLO (see the YOLO paper).
Thanks. wil check it out. so it seems like, it is possible to do
Yolo has grid method but InceptionV3 does not. MLMultiArray ` class TinyYOLOOutput : MLFeatureProvider { let grid: MLMultiArray
} `