我想保存Yolo或tensorflow的每次检测都存储在数组变量中以供进一步使用

Question

我想存储每个检测值并将其组合成一个数组以保存到excel中。我想截取sa_mask以将所有检测值添加到save_mask变量，但我尝试了截取，只得到了最后检测到的值。它不存储所有的值。输入图片描述

if FLAGS.count:
# count objects found
counted_classes = count_objects(pred_bbox, by_class = True)
# loop through dict and print
sa_counted = np.array(counted_classes)
print(sa_counted)

for key, value in counted_classes.items():
print("Number of {}s: {}".format(key, value))
sa_mask = ("{}  {}".format(key, value))
save_mask = []
save_mask.apped(sa_mask)

Answer 1

您只是没有存储值并将它们加在一起，所以每次进入循环时，它只是继续覆盖之前的结果…

我尽量简化一下:

# DATA
sa_counted1 = {"Surgical": 5, "KN95": 2, "Cloth": 1}
sa_counted2 = {"Surgical": 10, "KN95": 7, "Cloth": 3}
sa_counted3 = {"Surgical": 15, "KN95": 12, "Cloth": 5}
# WHERE THE RESULT WILL BE STORED
counted_classes = {}
# ADDING THE VALUES OF THE DICTIONARIES TOGETHER
for sa_counted in [sa_counted1, sa_counted2, sa_counted3]:
for key, value in sa_counted.items():
if key in counted_classes:
counted_classes[key] += value
else:
counted_classes[key] = value
# PRINT THE RESULT
for key, value in counted_classes.items():
print("Number of {}s: {}".format(key, value))

希望这对你有帮助。