資源簡介
Luna16數據集是三維的,使用yolov3進行肺結節檢測是無法直接處理的。需要把Luna16數據集的三維圖片轉換成二維圖片,把標注生成對應的.xml文件才行。getDataCsv.py生成.xml及二維圖片,getImg.py完成肺實質分割,getMat.py對疑似肺結節切割生成.Mat。 注意!!!!!原始getMat.py、traindataset.py有誤(有bug),因為csdn無法修改資源,詳情見https://blog.csdn.net/qq_24739717/article/details/101034728
代碼片段和文件信息
from?xml.etree.ElementTree?import?ElementElementTreetostring
import?jsoncsv
import?pandas?as?pd
from?getImg?import?load_itk_image?truncate_hu?normalazation?getLungMask
import?os
from?PIL?import?Image?ImageDraw
import?cv2
import?numpy?as?np
def?csvtoxml(fname):
????with?open(fname‘r‘)?as?f:
????????reader=csv.reader(f)
????????header=next(reader)
????????root=Element(‘Daaa‘)
????????print(‘root‘len(root))
????????for?row?in?reader:
????????????erow=Element(‘Row‘)
????????????root.append(erow)
????????????for?tagtext?in?zip(headerrow):
????????????????e=Element(tag)
????????????????e.text=text
????????????????erow.append(e)
????beatau(root)
????return?ElementTree(root)
?
?
?
def?beatau(elevel=0):
????if?len(e)>0:
????????e.text=‘\n‘+‘\t‘*(level+1)
?
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件????????9280??2019-09-19?13:11??Luna16數據集轉VOC數據集&肺實質分割&生成Mat\getDataCsv.py
?????文件????????9666??2019-09-17?22:30??Luna16數據集轉VOC數據集&肺實質分割&生成Mat\getImg.py
?????文件????????8077??2019-09-19?13:20??Luna16數據集轉VOC數據集&肺實質分割&生成Mat\getMat.py
?????目錄???????????0??2019-09-19?13:20??Luna16數據集轉VOC數據集&肺實質分割&生成Mat\
- 上一篇:仿朋友圈的一個
- 下一篇:Pytorch分類網絡肺結節假陽性剔除.zip
評論
共有 條評論