search

Tencent / PocketFlow

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.
https://pocketflow.github.io
Other
2.78k stars 490 forks source link

Does the PocketFlow support Inception V3 and YOLO V3 models? #266

Open mldlli opened 5 years ago

mldlli commented 5 years ago

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

jiaxiang-wu commented 5 years ago

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/