# The original Python program of Anuvani-Diarization 1.0

`diarize.py` and its package `anuvani_diar_ref/` are the program the release's published results come from:
- the DiariZen segmentation network and the WeSpeaker voiceprint network on PyTorch;
- then cannot-link VBx clustering and the learned decoder.

The pip package `anuvani_diar` (`../README.md`) runs the same pipeline on a Rust engine and gives the same bytes on
the setup the release's outputs were produced with. Use the package unless you need this exact code.

## Setting it up

Linux, an NVIDIA GPU, Python 3.10:

```bash
pip install torch==2.1.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121
git clone https://github.com/BUTSpeechFIT/DiariZen && cd DiariZen && git checkout 844f555
pip install -c constraints.txt -e . && pip install -e pyannote-audio && cd ..
pip install -r requirements.txt
```

`requirements.txt` lists these steps. `env/pip_freeze_reference.txt` is the full environment the outputs were verified
in (PyTorch 2.1.1 with CUDA 12.1). The program does not run in Google Colab, which cannot install PyTorch 2.1.1.

## Running it

```bash
python diarize.py --out out/ meeting.wav
python diarize.py --out out/ --in-dir recordings/
# the CPU works too, about 3× slower than real time
python diarize.py --device cpu --out out/ call.flac
```

- Each run writes `run.json`, with the weights' file hashes, the settings and an RTTM hash per file.
- If another `pyannote.audio` is installed in your user site-packages, run Python with `-s`, so that DiariZen's fork
  is the one imported.

## Install check

```bash
# the 12 recordings in ../golden/audio; expect "12/12 outputs identical"
python tools/verify_golden.py
```

A FAIL with matching audio usually means a different library version, GPU model, or non-default `--batch-size` or
`--threads`.

## Files

| Path | What |
|---|---|
| `diarize.py`, `anuvani_diar_ref/` | the program |
| `bundle/` | the weights as this program loads them (PyTorch checkpoints), the PLDA model and the decoder; `MANIFEST.json` lists every file's sha256 |
| `golden/expected.json` | the fingerprints of the 12 check recordings and of their RTTMs |
| `tools/verify_golden.py` | the install check |
| `requirements.txt`, `env/` | the environment |

The code is licensed under MIT OR Apache-2.0 (`../LICENSE-MIT`, `../LICENSE-APACHE`). The weights in `bundle/` are
the same as those in `../model/`: CC BY-NC 4.0 (the segmentation network) and CC BY 4.0 (the PLDA model and the
voiceprint network).