Metadata-Version: 2.4
Name: ThyroidAgent
Version: 0.1.1
Summary: DINOv3 + UNet for single-task segmentation and multi-task models (segmentation / malignancy classification / TIRADS), with dataset and training utilities
Author-email: wangbodong <wangbd8@mail2.sysu.edu.com>
License: MIT
Project-URL: Homepage, https://github.com/your-username/dino_unet_ori
Project-URL: Repository, https://github.com/your-username/dino_unet_ori
Keywords: segmentation,unet,dinov2,pytorch,medical imaging
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Image Recognition
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: torch>=1.12.0
Requires-Dist: torchvision>=0.13.0
Requires-Dist: timm>=0.6.0
Requires-Dist: numpy<2.0.0,>=1.20.0
Requires-Dist: Pillow>=8.0.0
Requires-Dist: scipy>=1.7.0
Provides-Extra: dev
Requires-Dist: build>=0.10.0; extra == "dev"
Requires-Dist: twine>=4.0.0; extra == "dev"
Requires-Dist: pytest>=7.0.0; extra == "dev"

# ThyroidAgent

ThyroidAgent is a PyTorch-based library that provides DINOv3 + UNet models for thyroid ultrasound image analysis, including both single-task segmentation and multi-task learning (segmentation + malignancy classification + TIRADS).

## Features

- **DINOv3_S_UNet**: single-task tumor segmentation.
- **DINOv3_S_UNet_MULTITASK**: joint segmentation, malignancy classification, and TIRADS prediction.
- **Dataset utilities**: `FullDataset` and `MultiTaskDataset` for training/validation pipelines.
- **Metrics and loss**: ready-to-use evaluation metrics and loss functions for medical image segmentation and multi-task learning.

## Installation

From PyPI (after you publish):

pip install ThyroidAgent
