Capture window - run yolov5 - show results (Windows only)
Project description
Capture window - run yolov5 - show results
pip install yolov5windowdetect
from yolov5windowdetect import Yolov5WindowDetect
winca = (
Yolov5WindowDetect(
pt_file=r"C:\Users\.....\best.pt",
repo_or_dir="./yolov5",
model="custom",
source="local",
)
.get_hwnd_window(window_title_regex=r"[Bb]lue[Ss]tacks.*")
.take_screenshot_and_run_yolov(
confidence_thresh=0.3,
show_results=True,
quit_key="q",
sleep_time=0.04,
rununtilstopped=True,
)
)
# window capture -> don't show results, only results as df
winca.get_hwnd_window(
window_title_regex=r"[Bb]lue[Ss]tacks.*"
).take_screenshot_and_run_yolov(
confidence_thresh=0.3, show_results=False, rununtilstopped=False,
).get_results_as_df()
# adb capture
winca.get_adb_window(
adb_path=r"C:\ProgramData\adb\adb.exe", adb_serial="localhost:5555",
).take_screenshot_and_run_yolov(
confidence_thresh=0.3, show_results=False, rununtilstopped=False,
).get_results_as_df()
Example
<a href="https://www.youtube.com/watch?v=-jXqL39Tf5w">
<img src="https://img.youtube.com/vi/-jXqL39Tf5w/0.jpg" style="width:100%;">
</a>
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for yolov5windowdetect-0.10-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b10ec04bf5cde7a3d11c5acf9ce8c07165788f3361232b06b1212135f960c58d |
|
MD5 | 671ca54344768682a66208e750a05f6b |
|
BLAKE2b-256 | 87559b9aeb5ae04cc58c349826b054ffcf07f1d7b75299cd625751335ab536f2 |