An-Labeler：AudioLabeller 高效音视频标注工具

An-Labeler：AudioLabeller 高效音视频标注工具详解（自研Qt）

一、工具概述

An Labeler 子类 AudioLabeller 是一款专为语音识别、声纹识别和音频分类任务设计的专业标注工具，支持多种音频格式和灵活的标注模式。

二、核心功能

1. 多格式支持

• 支持 WAV/MP3/FLAC/OGG 等常见音频格式
• 采样率自动转换（16kHz/44.1kHz/48kHz）
• 多声道音频分离标注
• 各类快捷键，加速标注过程

2. 标注模式

手动

3. 标注JSON

{
   
    "audio_file": "W:/Audio-SP-Projects/AudioPopDetect/PythonProject/AudioLabeler/d815927106414d5693f4c0a2c292c546.wav",
    "sample_rate": 16000,
    "duration": 3.34,
    "labels": [
        {
   
            "start": 1.8992748754966426,
            "end": 1.903895182725407,
            "label": "123"
        },
        {
   
            "start": 1.9118828325107284,
            "end": 1.9177952595577064,
            "label": "125"
        }
    ]
}

3. PY脚本（单文件）

https://github.com/Code-keys/AnLabeler

• AudioDemo

import os
import json
import numpy as np
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, 
                            QPushButton, QLabel, QFileDialog, QSlider, QMessageBox,
                            QInputDialog, QListWidget, QListWidgetItem, QSplitter, QStatusBar)
from PyQt5.QtCore import Qt, QTimer, QRectF, pyqtSignal
from PyQt5.QtGui import QPainter, QColor, QPen, QBrush
import pyqtgraph as pg
import soundfile as sf
import librosa
import librosa.display
from scipy.io import wavfile

import glob

def find_wav_files(folder_path):
    """使用glob查找所有WAV文件（包括子文件夹）"""
    return glob.glob(os.path.join(folder_path, '**/*.wav'), recursive=True)

class SyncViewBox(pg.ViewBox):
    """同步缩放视图"""
    def __init__(self, parent=None, *args, **kwargs):
        super().__init__(parent, *args, **kwargs)
        self.linked_views = []
    
    def linkView(self, view):
        if view not in self.linked_views:
            self.linked_views.append(view)

    def mouseDragEvent(self, ev, axis=None):
        super().mouseDragEvent(ev, axis)
        return
        if ev.isAccepted():
            for view in self.linked_views:
                view.setXRange(*self.viewRange()[0], padding=0)

    def mouseScrollEvent(self, ev, axis=None):
        """鼠标滚轮缩放时同步所有链接的ViewBox"""
        super().mouseScrollEvent(ev, axis)
        if ev.isAccepted():
            self._sync_linked_views()

    def _sync_linked_views(self):
        """同步所有链接的ViewBox的显示范围"""
        x_range, y_range = self.viewRange()
        for view in self.linked_views:
            # 避免递归调用（防止死循环）
            if view is not self:
                view.blockSignals(True)  # 临时阻止信号
                view.setRange(xRange=x_range, yRange=y_range, padding=0)
                view.blockSignals(False)

class AudioViewer(pg.PlotWidget):
    """音频视图基类"""
    selection_changed = pyqtSignal(float, float)
    
    def __init__(self, parent=None):
        view_box = SyncViewBox()
        super().__init__(parent, viewBox=view_box)
        self.setMouseEnabled(x=True, y=False)
        self.setLabel('bottom', 'Time (s)')

        self.hideAxis('left')
        self.plotItem.showGrid(x=True, y=True)

        self.setBackground(QColor(50, 50, 50))  # 暗灰色/ # 设置坐标轴和文本颜色为浅色（适合暗背景）
        self.getAxis('left').setPen('w')  # 白色 Y 轴
        self.getAxis('bottom').setPen('w')  # 白色 X 轴
        # 设置网格线（灰色半透明）
        self.showGrid(x=True, y=True, alpha=0.3)  # 显示网格，透明度 30%

        self.selection_start = None
        self.selection_end = None
        self.selection_rect = None
        self.selection_active = False
        self.playback_pos = None

    def linkView(self, view):
        self.getViewBox().linkView(view.getViewBox())

    def mousePressEvent(self, event):
        if event.button() == Qt.LeftButton:
            pos = self.plotItem.vb.mapSceneToView(event.pos())
            self.selection_start = pos.x()
            self.selection_end = pos.x()
            self.selection_active = True
            self.update_selection_rect()
        super().mousePressEvent(event)

    def mouseMoveEvent(self, event):
        if self.selection_active:
            pos = self.plotItem.vb.mapSceneToView(event.pos())
            self.selection_end = pos.x()
            self.update_selection_rect()
        super().mouseMoveEvent(event)

    def mouseReleaseEvent(self, event):
        if event.button() == Qt.LeftButton and self.selection_active:
            self.selection_active = False
            if abs(self.selection_end - self.selection_start) < 0.0000001:  # 点击而非拖动
                self.clear_selection()
            else:
                self.selection_changed.emit(*self.get_selection())
        super().mouseReleaseEvent(event)
    
    def update_selection_rect(self):
        if self.selection_start is not None and self.selection_end is not None:
            x0 = min(self.selection_start, self.selection_end)
            x1 = max(self.selection_start, self.selection_end)
            y_range = self.viewRange()[1]
            
            if self.selection_rect is None:
                self.selection_rect = pg.LinearRegionItem(values=[x0, x1])
                self.selection_rect.setZValue(10)
                self.selection_rect.setBrush(QBrush(QColor(0, 0, 255, 50)))
                self.addItem(self.selection_rect)
            else:
                self.selection_rect.setRegion([x0, x1])
    
    def clear_selection(self):
        self.selection_start = None
        self.selection_end = None
        if self.selection_rect:
            self.removeItem(self.selection_rect)
            self.selection_rect = None
    
    def get_selection(self):
        if self.selection_start is not None and self.selection_end is not None:
            return sorted([self.selection_start, self.selection_end])
        return None

    def set_playback_pos(self, pos):
        """设置播放位置指示器"""
        if self.playback_pos is None:
            self.playback_pos = pg.InfiniteLine(pos, angle=90, pen=pg.mkPen('r', width=2))
            self.addItem(self.playback_pos)
        else:
            self.playback_pos.setValue(pos)
    
    def clear_playback_pos(self):
        """清除播放位置指示器"""
        if self.playback_pos:
            self.removeItem(self.playback_pos)
            self.playback_pos = None

class WaveformViewer(AudioViewer):
    """波形视图"""
    def __init__(self, parent=None):
        super().__init__(parent)
        # 定义 Audition 风格的色
        # self.setLabel('top', 'Time (s)')
        audition_green = QColor(67, 217, 150)  # 纯青绿色
        self.waveform_plot = self.plot(
            pen=audition_green, width=1,
            symbol='o',  # 圆点符号
            symbolSize=4,  # 点的大小
            symbolBrush=audition_green,  # 填充颜色
            symbolPen=None  # 边框颜色（None 表示无边框）  
        )
        self.label_regio