Skip to content

Deepfaune

This is a Pytorch-Wildlife loader for the Deepfaune detector. The original Deepfaune model is available at: https://www.deepfaune.cnrs.fr/en/ Licence: CC BY-SA 4.0 Copyright CNRS 2024 simon.chamaille@cefe.cnrs.fr; vincent.miele@univ-lyon1.fr

DeepfauneDetector

Bases: YOLOV8Base

MegaDetectorV6 is a specialized class derived from the YOLOV8Base class that is specifically designed for detecting animals, persons, and vehicles.

Attributes:

Name Type Description
CLASS_NAMES dict

Mapping of class IDs to their respective names.
Source code in PytorchWildlife/models/detection/ultralytics_based/Deepfaune.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
class DeepfauneDetector(YOLOV8Base):
    """
    MegaDetectorV6 is a specialized class derived from the YOLOV8Base class 
    that is specifically designed for detecting animals, persons, and vehicles.

    Attributes:
        CLASS_NAMES (dict): Mapping of class IDs to their respective names.
    """

    CLASS_NAMES = {
        0: "animal",
        1: "person",
        2: "vehicle"
    }

    def __init__(self, weights=None, device="cpu"):
        """
        Initializes the MegaDetectorV5 model with the option to load pretrained weights.

        Args:
            weights (str, optional): Path to the weights file.
            device (str, optional): Device to load the model on (e.g., "cpu" or "cuda"). Default is "cpu".
        """
        self.IMAGE_SIZE = 960

        url = "https://pbil.univ-lyon1.fr/software/download/deepfaune/v1.3/deepfaune-yolov8s_960.pt" 
        self.MODEL_NAME = "deepfaune-yolov8s_960.pt"

        super(DeepfauneDetector, self).__init__(weights=weights, device=device, url=url)

__init__(weights=None, device='cpu')

Initializes the MegaDetectorV5 model with the option to load pretrained weights.

Parameters:

Name Type Description Default
weights str

Path to the weights file.

 None
device str

Device to load the model on (e.g., "cpu" or "cuda"). Default is "cpu".

 'cpu'
Source code in PytorchWildlife/models/detection/ultralytics_based/Deepfaune.py 
30
31
32
33
34
35
36
37
38
39
40
41
42
43
def __init__(self, weights=None, device="cpu"):
    """
    Initializes the MegaDetectorV5 model with the option to load pretrained weights.

    Args:
        weights (str, optional): Path to the weights file.
        device (str, optional): Device to load the model on (e.g., "cpu" or "cuda"). Default is "cpu".
    """
    self.IMAGE_SIZE = 960

    url = "https://pbil.univ-lyon1.fr/software/download/deepfaune/v1.3/deepfaune-yolov8s_960.pt" 
    self.MODEL_NAME = "deepfaune-yolov8s_960.pt"

    super(DeepfauneDetector, self).__init__(weights=weights, device=device, url=url)