tree-disk-segmentation

Tree Disk Segmentation

A Python package for analyzing tree rings in cross-sectional images.

Installation

pip install tree-disk-segmentation

Usage

Python API

import treedisksegmentation

# Configure the analyzer
treedisksegmentation.configure(
    input_image="input/tree-disk4.png",
    save_results=True,
)

# Run the analysis
(
    img_in,          # Original input image
    img_pre,         # Preprocessed image
    devernay_edges,  # Detected edges
    devernay_curves_f,  # Filtered curves
    devernay_curves_s,  # Smoothed curves
    devernay_curves_c,  # Connected curves
    devernay_curves_p,  # Final processed curves
) = treedisksegmentation.run()

Command Line Interface (CLI)

Basic usage:

tree-disk-segmentation --input_image ./input/baumscheibe.jpg --output_dir ./output/output.jpg

Save intermediate results:

tree-disk-segmentation --input_image ./input/baumscheibe.jpg --output_dir ./output/output.jpg --model_path ./models/u2net.pth --save_results

CLI Arguments

Argument	Type	Required	Default	Description
--input_image	str	Yes	-	Path to input image
--output_dir	str	No	./output	Output directory path
--model_path	str	No	./models/u2net.pth	Path to the pre-trained model weights
--debug	flag	No	False	Enable debug mode
--save_results	flag	No	False	Save intermediate images, labelme and config file

Development

Setting up Development Environment

1. Clone the repository:

git clone https://github.com/tuke307/tree-disk-segmentation.git
cd tree-disk-segmentation

1. Create and activate virtual environment:

poetry install
poetry shell