资源简介
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
评论
共有 条评论