Yolov5检测并生成文本及标签文件

最新推荐文章于 2024-06-10 01:18:13 发布

model2005

最新推荐文章于 2024-06-10 01:18:13 发布

阅读量4.6k

点赞数 4

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/model2005/article/details/121094483

python 同时被 3 个专栏收录

14 篇文章

订阅专栏

php

12 篇文章

订阅专栏

yolov5

11 篇文章

订阅专栏

本文介绍了一个从检测到标注的完整工作流程，包括使用Python脚本进行目标检测、将检测结果转换为文本文件、进一步将文本内容解析为数组及JSON格式，并最终生成XML标注文件的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

检测目标文件夹

python detect.py --source **path/**/ --weights **.pt --img 640

生成文本文件

加 --save-txt 参数即可

python detect.py --source **path/**/ --weights **.pt --img 640 --save-txt

文本文件转成数组

function fnTxt2Arr($fn,$iW,$iH){
$tStr = read2Str($fn);
$arr = explode("\r\n",$tStr);
$arr = array_filter($arr);//清除空字符
for($i=0;$i<count($arr);$i++){
$arrT = explode(" ",$arr[$i]);
   $tArr[$i]["labId"] = $arrT[0];//种类标签序号
$tArr[$i]["xMax"] = ($arrT[1]*$iW*2 + $arrT[3]*$iW )/2;
   $tArr[$i]["xMin"] = $arrT[1]*$iW*2 - $tArr[$i]["xMax"];
   $tArr[$i]["yMax"] = ($arrT[2]*$iH*2 + $arrT[4]*$iH )/2;
   $tArr[$i]["yMin"] = $arrT[2]*$iH*2 - $tArr[$i]["yMax"];
}
return $tArr;
}

返回种类标签，xmax,xmin,ymax,ymin的数组。

数组转成json字符串

function fnTxt2json($arr){//文本文件转成json
global $yEnArr, $arrImg, $arrColor;
for($i=0;$i<count($arr);$i++){
   $t1Arr[0]["x"] = $arr[$i]["xMin"];
   $t1Arr[0]["y"] = $arr[$i]["yMin"];
   $t1Arr[1]["x"] = $arr[$i]["xMax"];
   $t1Arr[1]["y"] = $arr[$i]["yMin"];
   $t1Arr[2]["x"] = $arr[$i]["xMax"];
   $t1Arr[2]["y"] = $arr[$i]["yMax"];
   $t1Arr[3]["x"] = $arr[$i]["xMin"];
   $t1Arr[3]["y"] = $arr[$i]["yMax"];
   $jArr[$i]["content"] = $t1Arr;
   $t2Arr["xMin"] = $arr[$i]["xMin"];
   $t2Arr["yMin"] = $arr[$i]["yMin"];
   $t2Arr["width"] = $arr[$i]["xMax"] - $arr[$i]["xMin"];
   $t2Arr["height"] = $arr[$i]["yMax"] - $arr[$i]["yMin"];
   $jArr[$i]["rectMask"] = $t2Arr;

   $t3Arr["labelName"] = trim($yEnArr[$arr[$i]["labId"]]);
   $tId = array_search($yEnArr[$arr[$i]["labId"]],$arrImg);
   $t3Arr["labelColor"] = $arrColor[$tId];
   $t3Arr["labelColorRGB"] = "undefined,undefined,undefined" ;
   $t3Arr["visibility"] = false; ;
   $jArr[$i]["labels"] = $t3Arr;

   $t4Arr["x"] = ($arr[$i]["xMin"] + $arr[$i]["xMax"])/2 ;
   $t4Arr["y"] = ($arr[$i]["yMin"] + $arr[$i]["yMax"])/2 ;
   $jArr[$i]["labelLocation"] = $t4Arr;
   $jArr[$i]["contentType"] = "rect";
}
return json_encode($jArr);
}