#选择night并生成json
import json
# 读取原始JSON文件
with open('D:/PycharmProjects/SODA10M/SSLAD-2D/labeled/annotations/instance_val.json', 'r') as f:
data = json.load(f)
new_data = {
"categories": data["categories"],
"annotations": [],
"images": []
}
# 遍历第三部分中的图片信息
for image_info in data["images"]:
if image_info["period"] == "Night":
image_id = image_info["id"]
file_name = image_info["file_name"]
# 根据image_id在第二部分中找到相应的标注信息,并添加到新的数据中
for annotation in data["annotations"]:
if annotation["image_id"] == image_id:
new_data["annotations"].append(annotation)
# 将满足条件的图片信息添加到新的数据中
new_data["images"].append(image_info)
# 将新的数据保存为新的JSON标签文件
with open('D:/PycharmProjects/SODA10M/SSLAD-2D/labeled/annotations/newnight.json', 'w') as f:
json.dump(new_data, f)
统计一下名字和数量
#从soda10m中选择照片为night
import json
# 读取json文件内容
with open('D:/PycharmProjects/SODA10M/SSLAD-2D/labeled/annotations/newnight.json', 'r') as f:
data = json.load(f)
night_images = []
count=0
# 遍历所有图片
for image in data['images']:
# 判断时段是否为Night,如果是则将文件名保存到night_images列表中
if image['period'] == 'Night':
night_images.append(image['file_name'])
count += 1
# 打印所有时段为Night的文件名
print(night_images)
print("时段为Night的图片数量:", count)