$ 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])