这个问题本质上是一个关于以根据您的训练数据排序的方式来表示您拥有的标签的最佳方式的问题。
表示
可以用于标签的简单表示可以是一个简单的 900 x 900 掩码(一个 numpy 数组),其中 1 和 0 表示相应像素是否具有该特征
顺序
假设您在连接输入图像和标签图像的文件名中有某种排序或标识符,您可以按文件名的顺序加载输入图像和标签图像:
def findpaths(path):
print(path)
im_paths = []
im_dict = {}
for root, dirs, files in os.walk(path, topdown=False):
# print(root, dirs, len(files))
for name in files:
if name.find('.png') != -1:
im_path = os.path.join(root, name)
im_paths.append(im_path)
im_id = int(re.findall('\d+', name)[0])
im_dict[im_path] = im_id
im_paths_sorted = sorted(im_paths, key=lambda x: im_dict[x])
return im_paths_sorted, im_dict
上面的示例显示了一个加载文件的函数,其命名约定为 d-1.png、d-2.png ... 等等。这将允许您按照文件名中的标识符的顺序创建文件路径列表。您可以对输入图像和标签图像执行此操作,然后按照标识符的顺序将它们维护为 numpy 数组。
希望这可以帮助!