darknet框架 训练分类模型及测试图片_darknet classifer命令

img
img

网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。

需要这份系统化的资料的朋友,可以戳这里获取

一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!

# 挂起后台训练命令
nohup ./darknet classifier train ../weather/weather.data ../weather/weather.cfg -gpus 0,1 -dont_show >../weather/train.log 2>&1 &

# 前台训练命令
nohup ./darknet classifier train ../weather/weather.data ../weather/weather.cfg -gpus 0,1 -dont_show >../weather/train.log 2>&1 &

2、darknet 分类模型单张图片检测命令

注意!!!
图片路径不正确的情况下,依旧会检测出结果!!!

# 单张图片预测命令
./darknet classifier predict /media/viatorsun/weather/weather.data /media/viatorsun/DataYZ/weather/weather.cfg /media/viatorsun/DataYZ/weather/weather.weights /media/viatorsun/weather/weather/7a964.jpg

检测结果

正确检测结果

错误检测结果

3、darknet 批量测试图片

通过darknet的 ./darknet classifier predict 命令只能对单张图片的进行测试,并且检测结果

# !/usr/bin/env python
# -\*- coding:utf-8 -\*-
# @Time : 2023.08
# @Author : 绿色羽毛
# @Email : lvseyumao@foxmail.com
# @Blog : https://blog.csdn.net/ViatorSun
# @Note : 将代码保存为 predict.py 放置到 /darknetAB3.0/predict.py


import os


data_path = '/media/viatorsun/DataSet/test'					# 待检测图像路径
save_dir  = "/media/viatorsun/DataSet/test\_out"				# 检测结果保存路径


if not os.path.exists(save_dir):
    os.makedirs(save_dir)

score = os.path.basename(data_path)

for img in os.listdir(data_path):

    img_path = os.path.join(data_path, img)

    cmd = 	'./darknet classifier predict ./myTest/0\_AI\_model/weather/weather.data ' \ 		
    		'./myTest/0\_AI\_model/weather/weather\_test.cfg ' \
          	'./myTest/0\_AI\_model/weather/weather\_100000.weights ' + img_path

    state = "".join(os.popen(cmd).readlines())
    lst = state.replace("\n\n", "\n").split("\n")

   
    if float(lst[-3].split(" ")[-1]) >= float(lst[-2].split(" ")[-1]) :
    	label = lst[-3].split(":")[0] + lst[-3].split(" ")[-1][:4]  
    else:
    	lst[-2].split(":")[0]



![img](https://img-blog.csdnimg.cn/img_convert/8a30a176fd5bb1ed6a75e6c3a1f6aec7.png)
![img](https://img-blog.csdnimg.cn/img_convert/e2d7375de6a4304cbf73ca09fa7e8d2f.png)

**网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。**

**[需要这份系统化的资料的朋友,可以戳这里获取](https://bbs.csdn.net/topics/618631832)**

**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**

ics/618631832)**

**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值