https://github.com/ds4sd/molclassifier
https://github.com/ds4sd/molclassifier
Last synced: over 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/ds4sd/molclassifier
- Owner: DS4SD
- License: mit
- Created: 2024-02-16T16:00:18.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2024-04-16T14:44:47.000Z (over 2 years ago)
- Last Synced: 2024-04-16T18:19:54.702Z (over 2 years ago)
- Language: Python
- Size: 275 KB
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
Awesome Lists containing this project
README
# MolClassifier
## Installation Instructions
- Ensure your distribution has the Python version 3.11:
```
python3 --version
pip3 -V
```
- Install poetry
```
pip install poetry
```
- Install the dependencies
```
poetry install
```
## Download the model
```
wget https://huggingface.co/ds4sd/MolClassifier/resolve/main/models/molclassifier_model.chpt -P ./data/models/
```
## Running the model
- Place the images to annotate in `./data/dataset/`
- Run the model
```
poetry run python ./mol_classifier/classifier.py infer --dir ./data/dataset/ --checkpoint ./data/models/molclassifier_model.chpt --output ./data/output.txt
```
- Read predictions in `./data/output.txt`