MobileNetV2-SSDLite代码分析-2 test

Github-pytorch-ssd
run_ssd_live_demo.py

作者逻辑异常清楚,就是创建好net(包括basenet+检测),送到Predictor里面去运行net和生成最终的结果

这边我运行的时候需要把GPU和CPU的变量调一下

from vision.ssd.vgg_ssd import create_vgg_ssd, create_vgg_ssd_predictor
from vision.ssd.mobilenetv1_ssd import create_mobilenetv1_ssd, create_mobilenetv1_ssd_predictor
from vision.ssd.mobilenetv1_ssd_lite import create_mobilenetv1_ssd_lite, create_mobilenetv1_ssd_lite_predictor
from vision.ssd.squeezenet_ssd_lite import create_squeezenet_ssd_lite, create_squeezenet_ssd_lite_predictor
from vision.ssd.mobilenet_v2_ssd_lite import create_mobilenetv2_ssd_lite, create_mobilenetv2_ssd_lite_predictor
from vision.utils.misc import Timer
import cv2
import sys
if len(sys.argv) < 4:
    print('Usage: python run_ssd_example.py    

你可能感兴趣的:(MobileNetV2-SSDLite代码分析-2 test)