PocketFlow
PocketFlow copied to clipboard
Does the PocketFlow support Inception V3 and YOLO V3 models?
I have two cases, one use the Inception V3 Regression model , and the other use the YOLO V3 model。 I noticed that the demos in Pocketflow are for LeNet or ResNet. I'm sure whether it supports Inception V3 and YOLO V3 and cann't find this in tutorials.
Can someone give me a suggestion?
Thanks
You can add support for your self-defined models (Inception-v3 for regression or YOLO-v3) by creating a new ModelHelper. See doc below: https://pocketflow.github.io/self_defined_models/