Spaces:
Running
Running
File size: 2,809 Bytes
1b58092 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import argparse
import json
import os
from PIL import Image, ImageDraw, ImageFont
from tqdm import tqdm
FOOT = ImageFont.truetype('/usr/share/fonts/dejavu/DejaVuSans-Bold.ttf', 50)
def custom_image(img_paths, save_path, image_size=448):
captions = ['CAM_FRONT_LEFT', 'CAM_FRONT', 'CAM_FRONT_RIGHT', 'CAM_BACK_LEFT', 'CAM_BACK', 'CAM_BACK_RIGHT']
width = image_size * 2
height = image_size
# count = 0
all_images = {}
for image_id, image_files in tqdm(img_paths.items()):
all_images[image_id] = dict()
all_images[image_id]['images_path'] = image_files
all_images[image_id]['images_size'] = {k: (0, 0) for k in image_files.keys()}
imgs = {}
for caption, image_file in image_files.items():
image_path = os.path.join(args.data_root, image_file.replace('../nuscenes/samples/', '/nuscenes/samples/'))
img = Image.open(image_path).convert('RGB')
old_wide, old_height = img.size
all_images[image_id]['images_size'][caption] = (old_wide, old_height)
img = img.resize((width, height))
draw = ImageDraw.Draw(img)
text = caption
draw.text((0, 0), text, fill=(255, 0, 255), font=FOOT)
imgs[caption] = img
result_width = width * 3
result_height = height * 2
result_img = Image.new('RGB', (result_width, result_height))
imgs = [imgs[caption] for caption in captions]
for i in range(len(imgs)):
row = i // 3
col = i % 3
left = col * width
top = row * height
right = left + width
bottom = top + height
result_img.paste(imgs[i], (left, top))
result_path = os.path.join(save_path, image_id + '.jpg')
result_img.save(result_path)
def get_images(ann_file):
with open(ann_file, 'r') as f: # , \
train_file = json.load(f)
images = {}
for scene_id in train_file.keys():
scene_data = train_file[scene_id]['key_frames']
for frame_id in scene_data.keys():
image_id = scene_id + '_' + frame_id
if image_id not in images:
images[image_id] = scene_data[frame_id]['image_paths']
else:
print(image_id)
return images
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--data-root', type=str, default='InternVL-Domain-Adaptation-Data/images/drivelm')
parser.add_argument('--ann-file', type=str, default='path/to/v1_1_val_nus_q_only.json')
args = parser.parse_args()
images = get_images(args.ann_file)
save_path = os.path.join(args.data_root, 'stitch')
os.makedirs(save_path, exist_ok=True)
custom_image(img_paths=images, save_path=save_path)
|