Skip to main content

Interactive coordinate picker for Computer Vision frameworks

Project description

pixpick ๐ŸŽฏ

Interactive coordinate picker for Computer Vision โ€” no external tools needed.

PyPI version Downloads Python 3.8+ License: MIT


The problem

Every major CV framework needs coordinates before it can run.

regioncounter = RegionCounter(region=[120, 80, 640, 480])    # YOLO   โ€” where does this region come from?
predictor.predict(box=np.array([120, 80, 640, 480]))         # SAM2   โ€” same problem

The standard workflow: open CVAT or Roboflow โ†’ grab coordinates โ†’ paste them back into code. Every. Single. Time.

The fix

import pixpick

region = pixpick.box("frame.jpg")      # drag a box on the image
zone   = pixpick.polygon("frame.jpg")  # click polygon vertices

# coordinates are ready โ€” unpack directly into any framework
# YOLO:
regioncounter = RegionCounter(
     region=region.yolo_region,  # pass region points
     model="yolo26n.pt",
 )

# same for YOLOE
model.predict("frame.jpg", visual_prompt= region.yolo_prompt())

# SAM1/SAM2:
predictor.predict(box=region.sam())

A window opens on your image. You interact. You get framework-ready coordinates back in Python. No round-trips.


Install

pip install pixpick

Selectors

Selector How to use Returns
pixpick.box() Left-click + drag Box
pixpick.polygon() Click vertices โ†’ Enter to confirm Polygon

Box controls โ€” drag to draw ยท R to reset ยท Esc to cancel

Polygon controls โ€” LMB add point ยท RMB undo ยท Z clear ยท Enter confirm ยท Esc cancel


Output formats

Every selection object carries all the formats you'll ever need.

# โ”€โ”€ Box โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€
region = pixpick.box("frame.jpg")

region.xyxy              # [x1, y1, x2, y2]            absolute pixels
region.xywh              # [x, y, w, h]                absolute pixels
region.norm_xywh         # [x, y, w, h]                0.0 โ€“ 1.0  โ† YOLO label format
region.center            # (cx, cy)
region.area              # pixelsยฒ


# โ”€โ”€ Polygon โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€โ”€
zone = pixpick.polygon("frame.jpg")

zone.points              # [(x0,y0), (x1,y1), ...]     absolute pixels
zone.as_numpy            # np.array shape (N, 2)
zone.norm                # [(x0n,y0n), ...]             0.0 โ€“ 1.0
zone.bbox                # โ†’ Box   tight bbox around the polygon
zone.npoints             # int

For more details, see Selectors.

Framework integration

Framework Selector Method
Ultralytics YOLOE โ€” visual prompt Box region.yolo_prompt()
Ultralytics YOLO โ€” region Box region.yolo_region()
SAM / SAM2 โ€” box prompt Box region.sam()
Any other format Box / Polygon region.to_raw()

Persistence

Pick once, reuse forever.

region.save("zone.json")
region = pixpick.load("zone.json")   # Box and Polygon both work

Production pattern โ€” pick interactively the first time, load on every subsequent run:

from pathlib import Path
import pixpick

ZONE = "config/count_zone.json"

zone = pixpick.load(ZONE) if Path(ZONE).exists() else pixpick.polygon("frame.jpg")
zone.save(ZONE)

Docs

๐Ÿš€ Getting Started Installation, first selection, controls
๐ŸŽฏ Selectors All properties and methods for Box and Polygon
๐Ÿ”Œ Framework Integration YOLO, SAM2 and more
๐Ÿ’พ Persistence Save, load, JSON schema
๐Ÿ—๏ธ Architecture How it's built and how to extend it
๐Ÿ—บ๏ธ Roadmap What's coming next

Contributing

We welcome contributions! Please open a GitHub issue or submit a pull request. For major changes, please open an issue first to discuss what you would like to change.

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

pixpick-0.1.1.tar.gz (15.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pixpick-0.1.1-py3-none-any.whl (13.8 kB view details)

Uploaded Python 3

File details

Details for the file pixpick-0.1.1.tar.gz.

File metadata

  • Download URL: pixpick-0.1.1.tar.gz
  • Upload date:
  • Size: 15.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.18

File hashes

Hashes for pixpick-0.1.1.tar.gz
Algorithm Hash digest
SHA256 8a3b7b7a6b3876b0734a806c62abdd8f08a176d8af29b11aa61ee3b775ce6b7c
MD5 7d7e2ab53784244463a803c4e9e1f00a
BLAKE2b-256 e4ba84197de9ab2638ae92bbcdcd8841e6dbad06773689565025671e057f1c4b

See more details on using hashes here.

File details

Details for the file pixpick-0.1.1-py3-none-any.whl.

File metadata

  • Download URL: pixpick-0.1.1-py3-none-any.whl
  • Upload date:
  • Size: 13.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.18

File hashes

Hashes for pixpick-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 daca03f9221bbcfaca4a3f6b76d314614348e603de52e747f685005b0c7b047b
MD5 b4b49162cd82fa3137f491dd07354008
BLAKE2b-256 1d9fee9b44c2d30c28e9e53b982beecf7f7ded390202a7e8530eff1290b1fe50

See more details on using hashes here.

Supported by

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