对YOLOv3模型调用时候的python接口详解
作者:NcepuKZH 时间:2023-05-31 00:58:57
需要注意的是:更改完源程序.c文件,需要对整个项目重新编译、make install,对已经生成的文件进行更新,类似于之前VS中在一个类中增加新函数重新编译封装dll,而python接口的调用主要使用的是libdarknet.so文件,其余在配置文件中的修改不必重新进行编译安装。
之前训练好的模型,在模型调用的时候,总是在
lib = CDLL("/home/*****/*******/darknet/libdarknet.so", RTLD_GLOBAL)这里读不到darknet编译生成的.so文件,导致直接的报错;之前以为是文件路径的问题,稀里糊涂的;由于很久不写c文件了,所以最后直接在python接口后在py文件中修改的画框、标置信度等操作,一次次的尝试后终于成功
(1)将项目中python文件下的darknet.py文件拷贝到根目录,和/libdarknet.so在同一个目录下
(2)整个demo程序都是用绝对路径;
实现yolov3模型加载,批量读取文件夹下的照片到库函数变量,最终处理结果存入在另外新建文件夹
###2019.04.03 by ylxb
def showPicResult(image,peoplecar,outimage):
img = cv2.imread(image)
out_img =outimage
cv2.imwrite(out_img, img)
for i in range(len(peoplecar)):
x1=peoplecar[i][2][0]-peoplecar[i][2][2]/2
y1=peoplecar[i][2][1]-peoplecar[i][2][3]/2
x2=peoplecar[i][2][0]+peoplecar[i][2][2]/2
y2=peoplecar[i][2][1]+peoplecar[i][2][3]/2
im = cv2.imread(out_img)
cv2.rectangle(im,(int(x1),int(y1)),(int(x2),int(y2)),(255,255,0),3)
text = listpeoplecar[i][0]
# 在图片上添加文字信息
if(text=="people"):
carcol=(55, 55, 255)#颜色显示
else:
carcol = (255, 55, 55)
cv2.putText(im, text, (int(x1), int(y1)), cv2.FONT_HERSHEY_SIMPLEX,
0.8, carcol, 1, cv2.LINE_AA)
#This is a method that works well.
cv2.imwrite(out_img, im)
###2019.04.03 by ylxb
filenames = os.listdir(picDir)
i = 0
num = 0#目标个数
car_num = 0#car个数
people_num = 0#people个数
car = "car" # car元素
people = "people" # people元素
for name in filenames:
filename=os.path.join(picDir,name)
#print(filename)
listpeoplecar = detect(net, meta, filename)
print(listpeoplecar)
i = i + 1
#save_picpath = out_img+str(filename).split("/")[-1].split(".")[0] + ".png"
out_img=out_img1+str(i)+'.png'
showPicResult(filename,listpeoplecar,out_img)
for item in listpeoplecar:
#print(item)
car_num = car_num + item[0].count(car)#car个数
people_num = people_num + item[0].count(people)#people个数
num = num + 1#目标个数
print('car个数: ' + str(car_num))
print('people个数: ' + str(people_num))
print('共检测出目标个数: ' + str(num))
print('共检测照片个数:'+ str(i))
放其中一个照片测试照片:
来源:https://blog.csdn.net/NcepuKZH/article/details/89151858
标签:YOLOv3,python
0
投稿
猜你喜欢
PHP数组交集的优化代码分析
2023-09-29 21:58:59
web.py获取上传文件名的正确方法
2021-01-22 16:00:06
Python爬虫辅助利器PyQuery模块的安装使用攻略
2023-10-18 02:19:34
动态提示的select下拉框
2007-12-02 14:54:00
详解Python字典的操作
2023-07-10 11:20:50
Go 语言数组和切片的区别详解
2024-04-25 15:08:37
详解MySQL主键唯一键重复插入解决方法
2024-01-20 16:41:22
Python 使用 prettytable 库打印表格美化输出功能
2021-10-25 08:42:54
PHP Session ID的实现原理与实例
2023-07-13 13:36:53
在asp.net中KindEditor编辑器的使用方法小结
2023-03-11 21:13:41
WEB2.0网页制作标准教程(5)head区的其他设置
2007-11-13 13:28:00
关于Thinkphp6的日志问题
2023-06-06 10:54:23
Python抓取京东图书评论数据
2023-10-16 09:26:37
django基础之数据库操作方法(详解)
2024-01-19 12:42:25
node基于express框架操作Mysql数据库的步骤
2024-01-22 11:09:54
三表左连接查询的sql语句写法
2024-01-15 01:54:21
python基于机器学习预测股票交易信号
2021-09-16 02:48:05
python判断文件是否存在,不存在就创建一个的实例
2022-04-29 02:28:55
深入了解MySQL的数据类型以及建库策略
2008-12-17 16:16:00
详解设计模式中的工厂方法模式在Python程序中的运用
2022-11-14 01:06:58