Error message here!

Hide Error message here!

忘记密码?

Error message here!

请输入正确邮箱

Hide Error message here!

密码丢失?请输入您的电子邮件地址。您将收到一个重设密码链接。

Error message here!

返回登录

Close

NiftyNet 数据预处理

范中豪 2019-02-23 22:01:00 阅读数:250 评论数:0 点赞数:0 收藏数:0

NiftyNet项目介绍

使用NiftyNet时,我们需要先将图像数据和标签进行一次简单的处理,得到对应的.csv文件。
对应文件格式为:
img.csv

image path
img_name img_path

label.csv

label path
img_label img_path

在此给出一个二分类的生成该文件的demo。首先,已经将两个类别的图片分别存储在两个文件夹中
demo

import pandas as pd
import os
# 生成 img.csv
list_img = []
list_path = []
img_path = 'C:\\Users\\fan\\Desktop\\demo\\train\\ad'
img_name = os.listdir(img_path)
for i, item in enumerate(img_name):
list_img.append(item)
list_path.append(img_path + "\\" + item)
img_path = "C:\\Users\\fan\\Desktop\\demo\\train\\cn"
img_name = os.listdir(img_path)
for i, item in enumerate(img_name):
list_img.append(item)
list_path.append(img_path + "\\" + item)
data_frame = pd.DataFrame({'image': list_img, 'path': list_path})
data_frame.to_csv('C:\\Users\\fan\\Desktop\\demo\\train\\img_path.csv', index=False)
# 生成label.csv
list_label_name = []
list_label_path = []
label_path = 'C:\\Users\\fan\\Desktop\\demo\\train\\ad'
label_name = os.listdir(label_path)
for j, elem in enumerate(label_name):
list_label_name.append(elem[0:2])
list_label_path.append(label_path + '\\' + elem)
label_path = 'C:\\Users\\fan\\Desktop\\demo\\train\\cn'
label_name = os.listdir(label_path)
for j, elem in enumerate(label_name):
list_label_name.append(elem[0:2])
list_label_path.append(label_path + '\\' + elem)
print(list_label_name)
label_dataframe = pd.DataFrame({'label': list_label_name, 'path': list_label_path})
label_dataframe.to_csv('C:\\Users\\fan\\Desktop\\demo\\train\\label.csv', index=False)

NiftyNet平台配置介绍

版权声明
本文为[范中豪]所创,转载请带上原文链接,感谢
https://www.cnblogs.com/zhhfan/p/10424489.html

编程之旅,人生之路,不止于编程,还有诗和远方。
阅代码原理,看框架知识,学企业实践;
赏诗词,读日记,踏人生之路,观世界之行;

支付宝红包,每日可领