gluoncv 用已經訓練好的模型參數,檢測物體


 當然這個模型參數,最好用自己的,否則不夠精確,我自己的還沒訓練完。

from matplotlib import pyplot as plt
import gluoncv
from gluoncv import model_zool_zoo,data,utils

net = model_zoo.get_model('faster_rcnn_resnet50_v1b_voc',pretrained = True)
x, orig_img = data.transforms.presets.rcnn.load_test('/mnt/hdfs-data-4/data/20180529_151800__00110383_01.jpg')
box_ids, scores, bboxes = net(x)
ax = utils.viz.plot_bbox(orig_img, bboxes[0], scores[0], box_ids[0], class_names=net.classes)

plt.show()


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM