创建包含图像的.csv文件

假设我有一个训练文件夹。这个训练文件夹包含多个单独的文件夹,每个文件夹里有图像(CNV、DME、DRUSEN 和 NORMAL 都是检测类型):

enter image description here

我想遍历每个文件夹中的每张图像,并为每张图像分配一个

  • 唯一ID疾病名称

然后将这些信息写入一个CSV文件

我希望达到的示例效果如下:

enter image description here

我希望有一个CSV文件,其中一列包含我拥有的60,000张图像的唯一ID,这些图像分布在各个文件夹中,另一列则包含疾病类型(或文件名:CNV、DME、DRUSEN、NORMAL)。


回答:

你可以使用os.listdir()来实现这个功能。

csv = 'id, disease\n'for disease in os.listdir(root_data_path):    for file in os.listdir(os.path.join(root_data_path, disease)):        csv += f'{file},{disease}\n'with open('diseases_ids.csv', 'w+') as output_csv:    output_csv.write(csv)

我还没有测试过这个代码,但你明白这个思路,对吗?

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注