在模型定义中,增加了移动端的前向传播过程。修改了toAndroid的代码。
This commit is contained in:
+2
-2
@@ -2,6 +2,6 @@
|
||||
食物分类网络模型包
|
||||
"""
|
||||
|
||||
from .food_net import FoodCNN, create_food_cnn
|
||||
from .food_net import FoodCNN, create_food_cnn,create_mobile_food_cnn
|
||||
|
||||
__all__ = ['FoodCNN', 'create_food_cnn']
|
||||
__all__ = ['FoodCNN', 'create_food_cnn','create_mobile_food_cnn']
|
||||
Reference in New Issue
Block a user