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 docs/base/data/datasets.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Datasets Module

::: PytorchWildlife.data.datasets
3 changes: 3 additions & 0 deletions docs/base/data/transforms.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Transforms Module

::: PytorchWildlife.data.transforms
3 changes: 3 additions & 0 deletions docs/base/models/classification/base_classifier.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Base Classifier

::: PytorchWildlife.models.classification.base_classifier
3 changes: 3 additions & 0 deletions docs/base/models/classification/resnet_base/amazon.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Amazon

::: PytorchWildlife.models.classification.resnet_base.amazon
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# ResNet Base

::: PytorchWildlife.models.classification.resnet_base.base_classifier
3 changes: 3 additions & 0 deletions docs/base/models/classification/resnet_base/custom_weights.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Custom Weights

::: PytorchWildlife.models.classification.resnet_base.custom_weights
3 changes: 3 additions & 0 deletions docs/base/models/classification/resnet_base/opossum.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Opossum

::: PytorchWildlife.models.classification.resnet_base.opossum
3 changes: 3 additions & 0 deletions docs/base/models/classification/resnet_base/serengeti.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Serengeti

::: PytorchWildlife.models.classification.resnet_base.serengeti
3 changes: 3 additions & 0 deletions docs/base/models/classification/timm_base/DFNE.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# DFNE

::: PytorchWildlife.models.classification.timm_base.DFNE
3 changes: 3 additions & 0 deletions docs/base/models/classification/timm_base/Deepfaune.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Deepfaune

::: PytorchWildlife.models.classification.timm_base.Deepfaune
3 changes: 3 additions & 0 deletions docs/base/models/classification/timm_base/base_classifier.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Timm Base

::: PytorchWildlife.models.classification.timm_base.base_classifier
3 changes: 3 additions & 0 deletions docs/base/models/detection/base_detector.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Base Detector

::: PytorchWildlife.models.detection.base_detector
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# HerdNet

::: PytorchWildlife.models.detection.localization.herdnet
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet/animaloc/data/patches.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Patches

::: PytorchWildlife.models.detection.localization.animaloc.data.patches
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet/animaloc/data/types.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Types

::: PytorchWildlife.models.detection.localization.animaloc.data.types
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet/animaloc/eval/lmds.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# LMDS

::: PytorchWildlife.models.detection.localization.animaloc.eval.lmds
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet/animaloc/eval/stitchers.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Stitchers

::: PytorchWildlife.models.detection.localization.animaloc.eval.stitchers
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet/dla.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# DLA

::: PytorchWildlife.models.detection.localization.dla
3 changes: 3 additions & 0 deletions docs/base/models/detection/herdnet/model.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Model

::: PytorchWildlife.models.detection.localization.model
3 changes: 3 additions & 0 deletions docs/base/models/detection/ultralytics_based/Deepfaune.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# Deepfaune

::: PytorchWildlife.models.detection.ultralytics_based.Deepfaune
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# MegaDetector v5

::: PytorchWildlife.models.detection.ultralytics_based.megadetectorv5
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# MegaDetector v6

::: PytorchWildlife.models.detection.ultralytics_based.megadetectorv6
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# MegaDetector v6 Distributed

::: PytorchWildlife.models.detection.ultralytics_based.megadetectorv6_distributed
3 changes: 3 additions & 0 deletions docs/base/models/detection/ultralytics_based/yolov5_base.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# YOLOv5 Base

::: PytorchWildlife.models.detection.ultralytics_based.yolov5_base
3 changes: 3 additions & 0 deletions docs/base/models/detection/ultralytics_based/yolov8_base.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# YOLOv8 Base

::: PytorchWildlife.models.detection.ultralytics_based.yolov8_base
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# YOLOv8 Distributed

::: PytorchWildlife.models.detection.ultralytics_based.yolov8_distributed
37 changes: 37 additions & 0 deletions docs/base/overview.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
# PytorchWildlife Base Module

The `PytorchWildlife` base module is the core component of this repository, designed to facilitate wildlife detection and classification tasks using PyTorch. It provides utilities for data processing, model implementation, and post-processing. It is also what is currently packaged in our Python package.

## Overview

The module is structured into the following submodules:

- **`data`**: Contains utilities for handling datasets and applying transformations.
- **`models`**: Includes implementations for classification and detection models.
- **`utils`**: Provides miscellaneous utilities for post-processing and other tasks.

## Submodules

### `data`
- `datasets.py`: Defines dataset classes for loading and preprocessing data.
- `transforms.py`: Implements data augmentation and transformation utilities.

### `models`
- `classification/`: Contains classification model architectures.
- `detection/`: Includes detection model architectures.

### `utils`
- `misc.py`: Provides helper functions for miscellaneous tasks.
- `post_process.py`: Implements post-processing utilities for model outputs.

## Getting Started

To use the `PytorchWildlife` module, import the required submodules as follows:

```python
from PytorchWildlife.data import datasets, transforms
from PytorchWildlife.models import classification, detection
from PytorchWildlife.utils import misc, post_process
```

Refer to the specific submodule documentation for detailed usage instructions.
1 change: 1 addition & 0 deletions docs/base/utils/misc.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
::: PytorchWildlife.utils.misc
1 change: 1 addition & 0 deletions docs/base/utils/post_process.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
::: PytorchWildlife.utils.post_process
10 changes: 10 additions & 0 deletions docs/fine_tuning_modules/classification/overview.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
---
description: "PyTorch-Wildlife classification fine-tuning: adapt species classifiers to your own datasets and geographic regions."
tags:
- PyTorch-Wildlife
- classification fine-tuning
- species classification
- transfer learning
---

# In Construction
10 changes: 10 additions & 0 deletions docs/fine_tuning_modules/detection/overview.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
---
description: "PyTorch-Wildlife detection fine-tuning: fine-tune MegaDetector on your own camera-trap data for improved local performance."
tags:
- PyTorch-Wildlife
- detection fine-tuning
- MegaDetector
- transfer learning
---

# In Construction
10 changes: 10 additions & 0 deletions docs/fine_tuning_modules/overview.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
---
description: "PyTorch-Wildlife fine-tuning modules: adapt MegaDetector and classification models to your own camera-trap datasets."
tags:
- PyTorch-Wildlife
- fine-tuning
- MegaDetector
- transfer learning
---

# In construction
62 changes: 62 additions & 0 deletions mkdocs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,62 @@ nav:
- Tags: tags.md
- Cite Us: cite.md
- Developer Guide: build_mkdocs.md
- Reference - Code API:
- Base Module:
- Overview: base/overview.md
- Data:
- Datasets: base/data/datasets.md
- Transforms: base/data/transforms.md
- Models:
- Classification:
- Base Classifier:
- base/models/classification/base_classifier.md
- ResNet Base:
- base/models/classification/resnet_base/base_classifier.md
- base/models/classification/resnet_base/amazon.md
- base/models/classification/resnet_base/custom_weights.md
- base/models/classification/resnet_base/opossum.md
- base/models/classification/resnet_base/serengeti.md
- TIMM Base:
- base/models/classification/timm_base/base_classifier.md
- base/models/classification/timm_base/DFNE.md
- base/models/classification/timm_base/Deepfaune.md
- Detection:
- Base Detector:
- base/models/detection/base_detector.md
- HerdNet:
- base/models/detection/herdnet.md
- base/models/detection/herdnet/dla.md
- base/models/detection/herdnet/model.md
- Animaloc:
- Data:
- base/models/detection/herdnet/animaloc/data/patches.md
- base/models/detection/herdnet/animaloc/data/types.md
- Eval:
- base/models/detection/herdnet/animaloc/eval/lmds.md
- base/models/detection/herdnet/animaloc/eval/stitchers.md
- Ultralytics Based:
- base/models/detection/ultralytics_based/Deepfaune.md
- MegaDetector v5:
- base/models/detection/ultralytics_based/megadetectorv5.md
- MegaDetector v6:
- base/models/detection/ultralytics_based/megadetectorv6.md
- base/models/detection/ultralytics_based/megadetectorv6_distributed.md
- YOLOv5 Base:
- base/models/detection/ultralytics_based/yolov5_base.md
- YOLOv8 Base:
- base/models/detection/ultralytics_based/yolov8_base.md
- YOLOv8 Distributed:
- base/models/detection/ultralytics_based/yolov8_distributed.md
- Utils:
- base/utils/misc.md
- base/utils/post_process.md
- Model Fine-tuning:
- Overview: fine_tuning_modules/overview.md
- Classification Fine-tuning:
- Overview: fine_tuning_modules/classification/overview.md
- Detection Fine-tuning:
- Overview: fine_tuning_modules/detection/overview.md

markdown_extensions:
- admonition
Expand Down Expand Up @@ -110,3 +166,9 @@ plugins:
type: date
enable_creation_date: true
fallback_to_build_date: true
- mkdocstrings:
handlers:
python:
paths: [src]
options:
docstring_style: google