根据x-ray生成训练csv文件

最新推荐文章于 2024-10-30 19:14:46 发布

范德彪陕西分彪

最新推荐文章于 2024-10-30 19:14:46 发布

阅读量2.1k

点赞数

分类专栏： pytorch学习文章标签：深度学习机器学习 python

本文链接：https://blog.csdn.net/weixin_46815330/article/details/122485447

版权

pytorch学习专栏收录该内容

138 篇文章 12 订阅

订阅专栏

本文介绍了一种使用Python构建图像数据集的方法。通过列出正常和肺炎两类图片文件，并为每张图片分配标签，最终形成可用于训练机器学习模型的数据集。示例中使用了os模块来获取文件列表，pandas库则用于创建DataFrame并输出数据集。

摘要由CSDN通过智能技术生成

import os
import pandas as pd
l_nor = os.listdir(train_nor_dir)
l_nor_label = [0 for i in l_nor]

l_pne = os.listdir(train_pne_dir)
l_pne_label = [1 for i in l_pne]


l_img = l_nor + l_pne
l_label = l_nor_label + l_pne_label


data = {"img": l_img, "l_label": l_label}
data_df = pd.DataFrame(data)

在这里插入图片描述

data_df.values

array([['NORMAL2-IM-0771-0001.jpeg', 0],
       ['NORMAL2-IM-1294-0001-0002.jpeg', 0],
       ['IM-0675-0001.jpeg', 0],
       ...,
       ['person931_virus_1592.jpeg', 1],
       ['person861_virus_1506.jpeg', 1],
       ['person1051_bacteria_2985.jpeg', 1]], dtype=object)