MegaDetector
This package is a pip-installable version of the support/inference code for MegaDetector, an object detection model that helps conservation biologists spend less time doing boring things with camera trap images. Complete documentation for this Python package is available at megadetector.readthedocs.io.
If you aren't looking for the Python package specifically, and you just want to learn more about what MegaDetector is all about, head over to the MegaDetector repo.
If you don't want to run MegaDetector, and you just want to use the utilities in this package - postprocessing, manipulating large volumes of camera trap images, etc. - you may want to check out the megadetector-utils package, which is identical to this one, but excludes all of the PyTorch/YOLO dependencies, and is thus approximately one zillion times smaller.
Installation
Install with:
pip install megadetector
MegaDetector model weights aren't downloaded at the time you install the package, but they will be (optionally) automatically downloaded the first time you run the model.
Package reference
See megadetector.readthedocs.io.
Examples of things you can do with this package
Run MegaDetector on one image and count the number of detections
from megadetector.utils import url_utils
from megadetector.visualization import visualization_utils as vis_utils
from megadetector.detection import run_detector
# This is the image at the bottom of this page, it has one animal in it
image_url = 'https://github.com/agentmorris/MegaDetector/raw/main/images/orinoquia-thumb-web.jpg'
temporary_filename = url_utils.download_url(image_url)
image = vis_utils.load_image(temporary_filename)
# This will automatically download MDv5a; you can also specify a filename.
model = run_detector.load_detector('MDV5A')
result = model.generate_detections_one_image(image)
detections_above_threshold = [d for d in result['detections'] if d['conf'] > 0.2]
print('Found {} detections above threshold'.format(len(detections_above_threshold)))
Run MegaDetector on a folder of images
from megadetector.detection.run_detector_batch import \
load_and_run_detector_batch, write_results_to_file
# Choose a folder to run MD on recursively, and an output file
image_folder = '/path/to/megadetector_test_images'
output_file = '/path/to/output/file.json'
# The package will automatically download whichever model you request; you
# can also specify a filename.
model_name = 'MDV5A'
# Run the model on all images in the folder
results = load_and_run_detector_batch(model_name, image_folder)
# Write results to a format that Timelapse and other downstream tools like
write_results_to_file(results,
output_file,
relative_path_base=image_folder,
detector_file=model_name)
Contact
Contact cameratraps@lila.science with questions.
Gratuitous animal picture
Image credit University of Minnesota, from the Orinoquía Camera Traps data set.
1 maintainer
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
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file megadetector-10.0.25.tar.gz.
File metadata
- Download URL: megadetector-10.0.25.tar.gz
- Upload date:
- Size: 654.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.11.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
844ad93e4c611b01cba0b6b4386defb626c89dc7f241c7dae91cd0711656d0fc
|
|
| MD5 |
409eb022928abbe8eb8cbd5bd2b00687
|
|
| BLAKE2b-256 |
fcfd645971472afe123012bb1af5882ae35b3cd4f01b7b76bd4716cced4ad5ba
|
File details
Details for the file megadetector-10.0.25-py3-none-any.whl.
File metadata
- Download URL: megadetector-10.0.25-py3-none-any.whl
- Upload date:
- Size: 733.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.11.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
daa36a06f5af30c1b0f08ecc3451ab9ad25d5aad503b03fa12fc22fad34d2204
|
|
| MD5 |
5a5fbfb1d8cb910c28d7615195e2d255
|
|
| BLAKE2b-256 |
47d85e0f8812bc228145a59fde33fababb82cdd9871569c09d5c7b64302f39f5
|
