Tree Disk Pith Detection

A Python package for analyzing tree rings in cross-sectional images. Originally forked from hmarichal93/apd.

Installation

pip install tree-disk-pith

Usage

Python API

import treediskpith

# Configure the analyzer
treediskpith.configure(
    input_image="input/tree-disk4.png",
    model_path="models/apd_dl.h5",
    save_results=True,
)

# Run the detection
(
    img_in,          # Original input image
    img_pre,         # Preprocessed image
    pith,  # Center of the tree disk
) = treediskpith.run()

Command Line Interface (CLI)

Basic usage:

tree-disk-pith --input_image ./input/tree-disk4-seg.jpg --model_path ./models/yolo11s-det-pith.pt

Advanced usage with custom parameters:

tree-disk-pith \
    --input_image input/tree-disk3.png \
    --model_path ./models/yolo11s-det-pith.pt \
    --output_dir custom_output/ \
    --new_shape 640 \
    --save_results \
    --debug

CLI Arguments

Development

Setting up Development Environment

1. Create and activate virtual environment:

poetry config virtualenvs.in-project true
poetry env use python

poetry install
eval $(poetry env activate)

2. Running tests:

pytest

3. fetch dataset

python fetch_dataset.py

4. Download pretrained model

python fetch_pretrained_model.py