资源算法ResNet_MobileNet_InceptionV3

ResNet_MobileNet_InceptionV3

2020-02-28 | |  68 |   0 |   0

Comparison of MobileNet, ResNet50, and InceptionV3 in Keras

This is a simple example of using ResNet, MobileNet and InceptionV3 from Keras to do object detection and classification tasks.

Demo

图片.png

ModelsMobileNetResNet50InceptionV3
1st guess'traffic_light', 0.99999177'nematode', 0.090073794'traffic_light', 1.0
2nd guess'pole', 6.4110359e-06'dishwasher', 0.042908493'maillot', 8.7838531e-10
3rd guess'flagpole', 4.8521628e-07'cleaver', 0.026148975'walking_stick', 5.6997368e-10

In this case, both MobileNet and InceptionV3 successfully recognize the traffic light, but the ResNet50 seems to be confused by it. Also, as for Model size, we can see from the Model Comparision section, and get to know that MobileNet has really small size. MobileNet can be a great choice for real-time applications.

Papers

1. Model Comparison

ModelSizeParametersDepth
MobileNet17 MB4,253,86488
ResNet5099 MB25,636,712168
InceptionV392 MB23,851,784159

2. How to run the code:

  1. Select the network by changing the choice value in main.py:

# 1: MobileNet, 2: ResNet50, 3: InceptionV3choice = 1
  1. Run the main.py:

python  main.py

3. Requirements:

  • keras>=2.0.8

  • tensorflow>=1.3.0 or tensorflow-gpu>=1.3.0

  • numpy>=1.13.1

  • python >= 3.5


上一篇:YOLOv3-use-MobileNet

下一篇:Mobilenet_Object_Detection

用户评价
全部评价

热门资源

  • TensorFlow-Course

    This repository aims to provide simple and read...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • mxnet_VanillaCNN

    This is a mxnet implementation of the Vanilla C...

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • Klukshu-Sockeye-...

    KLUKSHU SOCKEYE PROJECTS 2016 This repositor...