$ sudo apt install libpq-dev $ poetry add -D psycopg2 sqlalchemy
列名 | 型 | 説明 |
id | bigserial | PK |
img_id | bigint | FK。img番号 (numpy.ndarray 形式の画像データ) |
label | integer | imgのラベル (0:ねこ, 1:いぬ, 2:にんげん とか) |
purpose | integer | 学習(0)・テスト(1)種別 |
列名 | 型 | 説明 |
id | bigserial | PK |
np_array_bytes | bytea | numpy.ndarray を pickel でシリアライズして格納する |
cursor.execute(
"""
INSERT INTO numpy_arrays(np_array_bytes)
VALUES (%s)
RETURNING id;
""",
(pickle.dumps(some_array))
)
img_id = cu.fetchone()[0]
cursor.execute(
"""
INSERT INTO CIFER10(img_id, label, purpose)
VALUES (%s,%s,%s)
""",
(img_id, label, 0)
)
cursor.execute(
"""
SELECT numpy_arrays.nyumpy_array
FROM CIFER10, numpy_arrays
WHERE CIFER10.id=%s AND CIFER10.img_id = numpy_arrays.id
""",
(id)
)
some_array = pickle.loads(cursor.fetchone()[0])