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

发表时间：2021-05-11

在DarkNet模型中编译YOLO-V2和YOLO-V3

选择型号
下载所需文件
将图形导入到Relay
加载测试图像
在TVM运行时执行

本文是使用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