Model: https://huggingface.co/SicariusSicariiStuff/X-Ray_Alpha
clone:
git clone https://github.com/SicariusSicariiStuff/X-Ray_Vision.git
cd X-Ray_Vision/
Settings up venv, (tested for python 3.11, probably works with 3.10)
python3.11 -m venv env
source env/bin/activate
Install dependencies
pip install git+https://github.com/huggingface/[email protected]
pip install torch
pip install pillow
pip install accelerate
Usage: write your custom prompts in prompts.txt, each line is a new prompt, 3 prompts will result in 3 inferences. You can use the default prompt that is already in prompts.txt - it is recommended.
python xRay-Vision.py /path/to/model/ /dir/with/images/
The output will print to console, and export the results into a dir named after your image dir with suffix "_TXT"
So if you run:
python xRay-Vision.py /some_path/x-Ray_model/ /home/images/weird_cats/
Then results will be exported to:
/home/images/weird_cats_TXT/