DataIntelligenceCrew / data-selection

3 stars 0 forks source link

rough resnet to sqlite implementation #14

Closed jiwonac closed 2 years ago

jiwonac commented 2 years ago

I wrote a rough draft of sqlite.py which uses the functions defined in cifar.py (previously cifar_to_resnet.py) to obtain image lables, PIL representations, and feature vectors then saves that data into an SQLite database cifar.db.

The SQLite database has one table images with columns:

id INTEGER PRIMARY KEY,
label INTEGER NOT NULL,
vector BLOB NOT NULL,
pil BLOB NOT NULL

This is a draft. There are most likely bugs that I'll need to iron out tomorrow. I'm submitting a PR now in case @RosielQ wants to have a look at it.

@mundrapranay can merge the PR in a future commit.