如何使用正则表达式从图像文件名中获取参数到 CNN 预测器？

Question

我学会了如何通过照片预测人的年龄。 在这种情况下，年龄是文件名的第一个数字，他们使用正则表达式，如下所示：

import re
pattern = r'([^/]+)_\d+_\d+.jpg$'
p = re.compile(pattern)
r = p.search('16_0_0.jpg')
print(r.group(1))

而 output 是：

16

但就我而言，我需要预测第二个参数。 不是年龄，而是性别（0 为男性，1 为女性）。 在像“16_0_0.jpg”这样的图像文件名中，我需要采用第二个参数，即 0。在“16_1_0.jpg”中，我有一个 16 岁的女人，在“16_0_0.jpg”中，我有一个 16 岁的男人. 我怎样才能只接受第二个参数？

此外，此模式将用于 traindata：

train_data, test_data, preproc) = vis.images_from_fname(DATADIR, pattern = pattern, is_regression = True, random_state = 42)

Answer 1

您可以修改模式以捕获第二个参数：

import re
pattern = r'\d+_(\d+)_\d+.jpg$'
p = re.compile(pattern)
r = p.search('16_0_0.jpg')
print(r.group(1))

在此模式中，\d+ 匹配一个或多个数字，第二组括号 (\d+) 捕获图像文件名中的第二个参数。 最终的正则表达式.jpg$ 匹配文件名的扩展名。

使用此模式，当您运行以下代码时：

r = p.search('16_0_0.jpg')
print(r.group(1))

您将获得 output 0，这是图像文件名的第二个参数。 希望此信息对您有所帮助。

Answer 2

另一种可能对您有用的方法：

filename = “16_0_0.jpg”
data = filename.split(“_”)
age = data[0]
gender = data[1]