hollance / YOLO-CoreML-MPSNNGraph

Tiny YOLO for iOS implemented using CoreML but also using the new MPS graph API.
MIT License
933 stars 252 forks source link

is it possible to use multiple bounding box using Inceptionv3 model? #1

Closed steve21124 closed 7 years ago

steve21124 commented 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
}

} `

hollance commented 7 years ago

You might be able to train Inception-v3 the same way as YOLO (see the YOLO paper).

steve21124 commented 7 years ago

Thanks. wil check it out. so it seems like, it is possible to do