绘制训练数据的分布图

时间:2019-02-04 01:15:18

标签: python matplotlib python-os

我正在尝试将标签数据绘制为直方图。我在.txt文件示例中有以下数据。

78/1/2014/3ac218c0c6c378.jpg
78/1/2010/588feea1fcc809.jpg
78/1/2010/071c428717ceca.jpg
78/1/2010/4c119af3fdb787.jpg
78/1/2014/1453dbb876c191.jpg
78/1/2013/5a9cb4ee24047d.jpg
78/1/2012/106f7ea2150ff7.jpg
78/1/2012/6481ae22830294.jpg
78/1/2013/23bc02fcf393f8.jpg
78/1/2014/84919e1b067a85.jpg
78/1/2013/603cd9ca33d572.jpg
78/1/2012/f5a52935ee6fc2.jpg
78/1/2012/80e52b10388f9e.jpg

现在在这78个文件夹中,我总共有1000多张图片,但是在train.txt中,我使用的是13。如何绘制train.txt和实际数据的分布?

到目前为止,我一直想做的是

代码

import matplotlib.pyplot as plt
import os
train = open("D:/compCarsThesisData/data/train_test_split/classification/train.txt", "r")
path = "D:/compCarsThesisData/data/image/78/"

for root, _, files in os.walk(path):
        cdp = os.path.abspath(root)
        for f in files:
            name,ext = os.path.splitext(f)
            if ext == ".jpg":
               for x in train:
                   plt.hist(int(x.rsplit('/')[0]), int(path.rsplit('/')[4]))
plt.show()

您的帮助将不胜感激。

谢谢

0 个答案:

没有答案