Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,9 @@ repos:
name: docformatter
entry: docformatter
language: python
# pre-commit.ci currently provisions Python 3.14 for hook envs; DeepForest's
# tooling stack doesn't support that yet (e.g., `untokenize` build errors).
language_version: python3.12
additional_dependencies: [docformatter]
types: [python]
args: ["--in-place"]
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name = "deepforest"
version = "2.1.0"
description = "Platform for individual detection from airborne remote sensing including trees, birds, and livestock. Supports multiple detection models, adding models for species classification, and easy fine tuning to particular ecosystems."
readme = "README.md"
requires-python = ">=3.10,<3.15"
requires-python = ">=3.10,<3.14"
license = {text = "MIT"}
keywords = ["deep-learning", "forest", "ecology", "computer-vision"]
classifiers = [
Expand Down
5 changes: 2 additions & 3 deletions src/deepforest/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,6 @@
import torch
from omegaconf import DictConfig, OmegaConf
from PIL import Image

Image.MAX_IMAGE_PIXELS = None

from pytorch_lightning.callbacks import LearningRateMonitor
from pytorch_lightning.loggers import CSVLogger
from torch import optim
Expand All @@ -22,6 +19,8 @@
from deepforest.datasets import prediction, training
from deepforest.metrics import RecallPrecision

Image.MAX_IMAGE_PIXELS = None


class deepforest(pl.LightningModule):
"""DeepForest model for tree crown detection in RGB images.
Expand Down
Loading