Metadata-Version: 2.1
Name: om_simple_encoder
Version: 0.0.7
Summary: OM Simple Encoder
Home-page: https://www.soco.ai
Author: kyusonglee
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: Free for non-commercial use
Classifier: Operating System :: OS Independent
Description-Content-Type: text/markdown

# Om Simple Encoder
Om Simple Encoder is to fintune the query encoder. Currently it includes only query image encoder. 

# Install
```
pip install om_simple_encoder
```

# How to start training
```
python train.py --img_dir {image_root_path} --data {data json} --num_workers=15 --gpus=1 --batch_size=64 
```

# Data Format
```json
[ {"key1": ["/path/image1.jpg","/path/image2.jpg"], "key2":  ["/path/image1.jpg","/path/image2.jpg"]},...]
```

# Demo
```python
from om_simple_encoder.encoder import Encoder
h_enc = Encoder("brand_model_v1.ckpt")
encoded_embeddings = h_enc.images(["a.png"])
print (encoded_embeddings)
```


