TVM yolov3优化代码修改（编译运行OK）

最新推荐文章于 2025-10-08 18:25:05 发布

原创

最新推荐文章于 2025-10-08 18:25:05 发布 · 3k 阅读

0 ·

CC 4.0 BY-SA版权

吴建明

文章标签：

#目标检测 #计算机视觉 #深度学习

本文介绍了使用TVM对YOLOv3模型进行优化的代码修改过程，包括了`yolov3_quantize_sample.py`的详细步骤。通过nnvm库导入并测试YOLOv3模型，结合TVM进行编译和运行，确保在优化后能够正常工作。同时提供了相关代码仓库链接及参考资源。

TVM yolov3优化代码修改（编译运行OK）
yolov3_quantize_sample.py
在这里插入图片描述

附https://github.com/makihiro/tvm_yolov3_sample代码：
yolov3_quantize_sample.py
import nnvm
import nnvm.frontend.darknet
import nnvm.testing.yolo_detection
import nnvm.testing.darknet
import matplotlib.pyplot as plt
import numpy as np
import tvm
from tvm import rpc
import sys
import cv2
import time

from ctypes import *
from tvm.contrib import util
from tvm.contrib.download import download
from nnvm.testing.darknet import __darknetffi__


# Model name
MODEL_NAME = 'yolov3'


######################################################################
# Download required files
# -----------------------
# Download cfg and weights file if first time.
CFG_NAME = MODEL_NAME + '.cfg'
WEIGHTS_NAME = MODEL_NAME + '.weights'
REPO_URL = 'https://github.com/siju-samuel/darknet/blob/master/'
CFG_URL = REPO_URL + 'cfg/' + CFG_NAME + '?raw=true'
WEIGHTS_URL = 'https://pjreddie.com/media/files/' + WEIGHTS_NAME


download(CFG_URL, CFG_NAME)
download(WEIGHTS_URL, WEIGHTS_NAME)


# Download and Loa