-
大小: 4KB文件類型: .rar金幣: 1下載: 0 次發(fā)布日期: 2021-01-09
- 語言: 其他
- 標(biāo)簽: YOLO??tfrecord??VOC??數(shù)據(jù)轉(zhuǎn)換??深度學(xué)習(xí)??
資源簡(jiǎn)介
將標(biāo)注好的YOLO格式數(shù)據(jù)轉(zhuǎn)換成VOC數(shù)據(jù)格式,并將VOC數(shù)據(jù)格式轉(zhuǎn)換成TFRECORDS格式,以便其他深度學(xué)習(xí)網(wǎng)絡(luò)使用。
代碼片段和文件信息
import?os
import?sys
import?random
?
import?numpy?as?np
import?tensorflow?as?tf
?
import?xml.etree.ElementTree?as?ET?#操作xml文件
?
#我的標(biāo)簽定義只有兩類,根據(jù)自己的圖片而定
VOC_LABELS?=?{
????‘TCFBA‘:?(0?‘TCFBA‘)
????‘TTFBS‘:?(1?‘TTFBS‘)
}
?
#?圖片和標(biāo)簽存放的文件夾.
DIRECTORY_ANNOTATIONS?=?‘Annotations/‘
DIRECTORY_IMAGES?=?‘JPEGImages/‘
?
#?隨機(jī)種子.
RANDOM_SEED?=?4242??
SAMPLES_PER_FILES?=?200??#每個(gè)文件的樣本數(shù)
?
#生成整數(shù)型,浮點(diǎn)型和字符串型的屬性
def?int64_feature(value):
????
????if?not?isinstance(value?list):
????????value?=?[value]
????return?tf.train.Feature(int64_list=tf.train.Int64List(value=value))
?
def?float_feature(value):
????
????if?not?isinstance(value?list):
????????value?=?[value]
????return?tf.train.Feature(float_list=tf.train.FloatList(value=value))
?
def?bytes_feature(value):
????
????i
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????文件???????6025??2018-08-15?14:34??VOC_tfrecord.py
?????文件???????6203??2018-08-15?15:17??yolo2voc.py
?????文件????????220??2018-08-15?15:24??README.txt
-----------?---------??----------?-----??----
????????????????12448????????????????????3
評(píng)論
共有 條評(píng)論