Skip to main content

Accurate and efficient spot detection with CNNs

Project description

PyPI - Python Version PyPI - Version PyPI - License PyPI - Status tests PyPI - Downloads


Spotipy - Accurate and efficient spot detection with CNNs


Install the correct tensorflow for your CUDA version.

Clone the repo and install it

git clone
pip install spotipy


A SpotNet spot detection model can be instantiated from a custom Config class:

from spotipy.model import Config, SpotNet

config = Config(

model = SpotNet(config,name="mymodel", basedir="models")


The training data for a SpotNet model consists of input image X and spot coordinates P (in y,x order):

import numpy as np
from spotipy.utils import points_to_prob

# generate some dummy data 
def dummy_data(n_samples=16):
    X = np.random.uniform(0,1,(n_samples, 128, 128))
    P = np.random.randint(0,128,(n_samples, 21, 2))
    for x, p in zip(X, P):
        x[tuple(p.T.tolist())] = np.random.uniform(2,5,len(p))
    Y = np.stack(tuple(points_to_prob(p[:,::-1], (128,128)) for p in P))
    return X, Y

X,Y = dummy_data(128)
Xv,Yv = dummy_data(16)

model.train(X,Y, validation_data=[X, Y], epochs=10, steps_per_epoch=128)



Applying a trained SpotNet:

img = dummy_data(1)[0][0]

prob, points = model.predict(img)


Albert Dominguez Mantes, Antonio Herrera, Irina Khven, Anjali Schläppi, Gioele La Manno, Martin Weigert

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

spotipy-detector-0.1.0.tar.gz (48.6 kB view hashes)

Uploaded Source

Built Distribution

spotipy_detector-0.1.0-py3-none-any.whl (22.0 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page