資源簡介
將標注好的YOLO格式數據轉換成VOC數據格式,并將VOC數據格式轉換成TFRECORDS格式,以便其他深度學習網絡使用。
代碼片段和文件信息
import?os
import?sys
import?random
?
import?numpy?as?np
import?tensorflow?as?tf
?
import?xml.etree.ElementTree?as?ET?#操作xml文件
?
#我的標簽定義只有兩類,根據自己的圖片而定
VOC_LABELS?=?{
????‘TCFBA‘:?(0?‘TCFBA‘)
????‘TTFBS‘:?(1?‘TTFBS‘)
}
?
#?圖片和標簽存放的文件夾.
DIRECTORY_ANNOTATIONS?=?‘Annotations/‘
DIRECTORY_IMAGES?=?‘JPEGImages/‘
?
#?隨機種子.
RANDOM_SEED?=?4242??
SAMPLES_PER_FILES?=?200??#每個文件的樣本數
?
#生成整數型,浮點型和字符串型的屬性
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
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件???????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
- 上一篇:adrcdemo.zip
- 下一篇:深信服虛擬化高級考題--A卷word里圖片
評論
共有 條評論