semvec/pypi-packages huggingface dataset:
rob-pitch 0.1.2
Robust pitch prediction using PyTorch
Requires Python: >=3.6

# RobPitch

## Overview

RobPitch is a pitch detection model trained to be robust against noise and reverberation environments. The model has been trained on 1600 hours of high-quality data, supplemented by an equivalent amount of simulated noisy and reverberant data, ensuring effective performance under challenging acoustic conditions.

## Installation via pip

Install RobPitch using the following command:

```
pip install rob-pitch==0.1.2
```

### Example of Usage

```
import robpitch

# Init model
model = robpitch.load_model()

# process the audio
outputs = model.infer("path/to/audio")
pitch = outputs['pitch']
latent_feature = outputs['latent']

```

## Local Setup

### Model Download

- Download the model from ![model](https://modelscope.ai/models/pandamq/robpitch-16k)

### Example of Local Usage

```
import torch
import numpy as np

from robpitch import RobPitch
from utils.audio import load_audio

# Initialize the model
robpitch = RobPitch()
device = torch.device("cpu")

# Load model from checkpoint
model = robpitch.load_from_checkpoint(
    config_path="config.yaml",
    ckpt_path="model.bin",
    device=device
)

# Load and process the audio
wav = load_audio(
    "path/to/audio",
    sampling_rate=16000,
    volume_normalize=True
)
wav = torch.from_numpy(wav).unsqueeze(0).float().to(device)

# Get model outputs
outputs = model(wav)
pitch = outputs['pitch']
latent_feature = outputs['latent']

```

For more detailed usage examples, refer to the ![exp/demo.ipynb](exp/demo.ipynb) notebook.