TVM在Darknet模型中编译yo-v2和yo-v3,DarkNet,YOLOV2,YOLOV3

发表时间:2021-05-11


本文是使用TVM部署Darknet模型的入门教程。所有必需的模型和库都将通过脚本从Internet下载。该脚本运行带有边界框的YOLO-V2和YOLO-V3模型。Darknet解析与CFFI和CV2库有依赖性。请在执行此脚本之前安装CFFI和CV2

pip install cffi
pip install opencv-python
# numpy and matplotlib
import numpy as np
import matplotlib.pyplot as plt
import sys

# tvm,relay
import tvm
from tvm import te
from tvm import relay
from ctypes import *
from tvm.contrib.download import download_testdata
from tvm.relay.testing.darknet import __darknetffi__
import tvm.relay.testing.yolo_detection
import tvm.relay.testing.darknet

选择型号

型号为:“ yolov2”,“ yolov3”或“ yolov3-tiny”

# 选择模型
# 'yolov2', 'yolov3' o

文章来源互联网,如有侵权,请联系管理员删除。邮箱:417803890@qq.com / QQ:417803890

微配音

Python Free

邮箱:417803890@qq.com
QQ:417803890

皖ICP备19001818号-4
© 2019 copyright www.pythonf.cn - All rights reserved

微信扫一扫关注公众号:

联系方式

Python Free