You can deploy this model and start earning money today!

andrewromanenco/
hit-detector

Name: hit-detector
Availability: InStock

Views

No views yet

🎯 Hit Detector Model

This PyTorch-based CNN detects holes on boards or paper using a sliding window approach. It was trained on image patches of size 24×24. The model scans larger images with this patch size to detect regions of interest.

Holes or defects must approximately fit within a 20×20 region to be accurately detected.

🤗 Model Card on Hugging Face.

🔗 Live Demo

Try the model here: Hit Detector Gradio Demo

📥 Model Inputs & Outputs

Input: RGB or grayscale image (PIL.Image)
Output: Annotated PIL.Image with red (or specified color) squares highlighting detected holes

🚀 Quick Start

🧠 Inference in Python

python

from PIL import Image
from  pipeline  import  HitDetectorPipeline

pipe  =  HitDetectorPipeline("model.pt")

img  = Image.open("input.png")
result  =  pipe(img)
result.save("output.png")
print("✅ Output saved to output.png")

📦 Installation

pip install -r requirements.txt

🧪 Testing in Docker

To test the model or pipeline scripts inside a clean container:

bash

cd <project folder>
docker run -it --rm -p 7860:7860 -v $PWD:/appx:rw romanenco/python-tool-chain /bin/bash
cd /appx
pip install -r requirements.txt
python test_pipeline.py

You should see output.png generated as a result.

🌐 Run Gradio UI