选择合适的NN模型(速度/性能)

时间:2019-04-05 11:04:12

标签: python machine-learning neural-network

我是初学者,这是我的第一步。

我已经在学习不同的神经网络体系结构,我有一个问题:

我应该为Rasberry PI / android选择哪种型号?

我已经尝试使用分辨率为98x98的“ ResNet”,该型号几乎需要PC的全部功能。完全是:

该型号需要2 GB的视频卡内存,1.4〜2 GB的RAM。

此型号不适用于android / Rasberry(低功耗)。

我应该为任务选择哪种型号?

P.S我希望在Rasberry上至少5〜10 FPS,在Android上至少10〜15 FPS。

2 个答案:

答案 0 :(得分:3)

在具有5-10FPS的Raspberry Pi上进行对象检测非常不现实。

您可以看一下YOLO或SSD,例如YOLO还有一个较小的实现,可以在RPI上运行,但您会对1FPS感到满意。

答案 1 :(得分:2)

TinyYOLO是原始YOLO网络的较小版本。您可以尝试那个。