platform-detector

Model that indicates platform with green stickers on it on an image

Prerequisites

python3 git tqdm torch torchvision

You should also have MPS installed if you're using MacOS silicon GPU. To enable MPS support, run

pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

Run code below in project folder:

python3 src/platform_detector.py