COCO KeyPoints關(guān)鍵點(diǎn)數(shù)據(jù)集準(zhǔn)備

概述

網(wǎng)上搜了一圈，coco關(guān)鍵點(diǎn)數(shù)據(jù)集準(zhǔn)備的內(nèi)容比較少，這里寫一篇完成的標(biāo)注流程到數(shù)據(jù)集準(zhǔn)備的文章，以備后忘

標(biāo)注工具

coco官方標(biāo)注工具: coco–annotator
https://github.com/jsbroks/coco-annotator

標(biāo)注過(guò)程

這里注意category_id 最好為1 ，否則會(huì)有不必要的麻煩 “category_id”: 1

標(biāo)注流程

1、創(chuàng)建類別
2、 創(chuàng)建關(guān)鍵點(diǎn)名稱
3、創(chuàng)建關(guān)鍵點(diǎn)之間連線關(guān)系
4、選擇一張圖片
5、標(biāo)注待識(shí)別目標(biāo)識(shí)別矩形框
6、在框內(nèi)標(biāo)注關(guān)鍵點(diǎn)keyPoints
7、標(biāo)注下一張圖片

標(biāo)注完成之后， 導(dǎo)出數(shù)據(jù)集
導(dǎo)出結(jié)果在 coco–annotator/datasets/數(shù)據(jù)集名字/. exports/ 文件夾里


ps: 數(shù)據(jù)集也可以直接在web頁(yè)面上點(diǎn)擊下載

數(shù)據(jù)集準(zhǔn)備

完善數(shù)據(jù)集

coco數(shù)據(jù)集標(biāo)準(zhǔn)字段

{"info": info,"licenses": [license],"images": [image],"annotations": [annotation],
}

導(dǎo)出的數(shù)據(jù)集默認(rèn)是沒(méi)有 info字段和licenses 字段的，所以我們?cè)跀?shù)據(jù)集的json文件（coco-1613789462.5181398.json）自己加上這兩個(gè)字段， 如下:

字段如下：

"info": {"contributor": "zyl","date_created": "2021/02/01","description": "human hand keypoints","url": "","version": "1.0","year": 2021},"licenses": [{"id": 1,"name": "Attribution-NonCommercial-ShareAlike License","url": "http://creativecommons.org/licenses/by-nc-sa/2.0/"}]

分割數(shù)據(jù)集為訓(xùn)練集和測(cè)試集

如下，三個(gè)文件放同一個(gè)目錄

修改splitCoco.sh 文件中的文件名為自己的json標(biāo)注文件路徑,運(yùn)行

./splitCoco.sh

執(zhí)行 得到訓(xùn)練集文件和測(cè)試集文件

test.json
train.json

圖片文件訓(xùn)練集和驗(yàn)證集區(qū)分

我們標(biāo)注的圖像是放在一個(gè)文件夾里的，這里需要區(qū)分成兩個(gè)文件夾 test2017/ 和 train2017/

ps:當(dāng)然這里最簡(jiǎn)單的方法是把所有的圖像分別拷貝到這兩個(gè)目錄， 即 test2017/ 和 train2017/ 里面都放所有的標(biāo)注圖像

分類可以通過(guò)pickUsedImg.py這個(gè)腳本進(jìn)行分類，自己改改，這個(gè)代碼看不懂這個(gè)文章也就不用往下看了。

最終得到的數(shù)據(jù)集

三個(gè)文件夾 annotations/ 、 test2017/ 、 train2017/
test2017 和train2017 分別是測(cè)試集和訓(xùn)練集圖片
其中 annotations/文件夾如下:

計(jì)算數(shù)據(jù)集的mean 和std 值

參照dataset_mean_std.py 文件 路徑修改為自己的數(shù)據(jù)集圖片全集的目錄
計(jì)算出的值如下：

到此，數(shù)據(jù)集準(zhǔn)備完畢

數(shù)據(jù)集處理所需代碼文件

鏈接: https://pan.baidu.com/s/1CmnnkthOXIjY4ndSRiqQ0g 提取碼: cai9 復(fù)制這段內(nèi)容后打開百度網(wǎng)盤手機(jī)App，操作更方便哦
–來(lái)自百度網(wǎng)盤超級(jí)會(huì)員v1的分享

總結(jié)

以上是生活随笔為你收集整理的COCO KeyPoints关键点数据集准备的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。