import json
import os
import subprocess
with open('/home/yunchuan/L2D-master/data/coco2014/train_anno.json', 'rb') as f:
images_info = json.load(f)
# 指定包含图像的目录路径
image_directory = '/home/yunchuan/L2D-master/data/coco2014/train2014/'
# 遍历字典,检查每个文件是否存在
missing_files = []
for image in images_info:
file_path = os.path.join(image_directory, image['file_name'])
if not os.path.exists(file_path):
missing_files.append(image['file_name'])
# 输出缺失的文件名
print("缺失的文件有:")
for file_name in missing_files:
print(file_name)
# 指定下载文件的目录
download_directory = '/home/yunchuan/L2D-master/data/coco2014/lose_images'
os.makedirs(download_directory, exist_ok=True)
# COCO数据集的基础URL
base_url = 'http://images.cocodataset.org/train2014/'
# 遍历缺失文件并下载
for file_name in missing_files:
# 构建完整的下载URL
download_url = f'{base_url}{file_name}'
# 构建wget命令
wget_command = f'wget -P {download_directory} {download_url}'
# 执行wget命令
subprocess.run(wget_command, shell=True)
# 输出已完成下载的信息
print("下载完成")
coco数据集train2014图片缺失问题
最新推荐文章于 2024-07-11 20:27:00 发布