File size: 1,306 Bytes
7c5440e |
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 |
from PIL import Image
import io
import os
import time
import requests
import msgpack
import base64
import numpy as np
from io import BytesIO
def select_frames(input_frames, num_segments = 10):
indices = np.linspace(start=0, stop=len(input_frames)-1, num=num_segments).astype(int)
frames = [input_frames[ind] for ind in indices]
return frames
tic = time.time()
video_dir = './images/'
frames = [(os.path.splitext(item)[0], os.path.join(video_dir, item)) for item in os.listdir(video_dir)]
frames = [item[1] for item in sorted(frames, key=lambda x: x[0])]
image_paths = select_frames(frames, num_segments=10)
request = {}
byte_images = []
for image_path in image_paths:
img = Image.open(image_path)
byte_io = BytesIO()
img.save(byte_io, format='PNG')
encoded_image = base64.b64encode(byte_io.getvalue()).decode('utf-8')
byte_images.append(encoded_image)
# Step 4: Pack the byte data with msgpack
#packed_data = msgpack.packb(byte_images)
payload = {
"images": byte_images,
}
IP = '10.100.193.192'
url = f'http://{IP}:8001/generate'
headers = {'Content-Type': 'application/json'}
tic1 = time.time()
response = requests.post(url, json=payload, headers=headers)
toc = time.time()
print(response.text)
print('stage 1', tic1 - tic)
print('stage 2', toc - tic1) |