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

model2005

关注

原创最新推荐文章于 2026-05-08 07:08:43 发布 4.7k 阅读 AI 写同款· GEO 优化 ›

Yolov5 多边形标签转换，所有json文件自动转成txt格式[详细过程] 注：labelme是麻省理工（MIT）的计算机科学和人工智能实验室（CSAIL）研发的图像标注工具，人们可以使用该工具创建定制化标注任务或执行图像标注，项目源代码已经开源。通过labelme对图进行标注后，得到的是json文件，而Yolov5对数据进行模型构建的时候，读取需要的是txt格式的文件。txt_outer_path——保存txt文本的文件夹的绝对路径。json_name——json_name是json文本的名字。json_floder——读取json的文件夹的绝对路径。阅读详情

检测目标文件夹

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);
}