Spaces:
Runtime error
Runtime error
File size: 790 Bytes
a56642d |
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 |
import json
from psycopg2.extras import RealDictCursor
#import cv2
import psycopg2
import cv2
CONNECTION = "postgres://postgres:"
conn = psycopg2.connect(CONNECTION)
cursor = conn.cursor(cursor_factory=RealDictCursor)
def get_sample():
camera_name, camera_id = 'cam2', 4
print('Executing SQL command')
cursor.execute("SELECT * FROM annotations WHERE camera_id = {} and time >='2021-05-01 00:00:00' and time <='2021-05-07 23:59:50' and label_id in (1,2)".format(camera_id))
print('Dumping to json')
annotations = json.dumps(cursor.fetchall(), indent=2, default=str)
wjdata = json.loads(annotations)
with open('{}_{}_test.json'.format(camera_name, camera_id), 'w') as f:
json.dump(wjdata, f)
print('Done dumping to json')
get_sample()
|