瀏覽代碼

refactor: fix path/source/feature_extractor

main
Rob Hallam 2 月之前
父節點
當前提交
01369f760e
共有 1 個文件被更改,包括 9 次插入2 次删除
  1. +9
    -2
      pipeline/feature_extractors.py

+ 9
- 2
pipeline/feature_extractors.py 查看文件

@@ -5,6 +5,8 @@ import random
import subprocess
from ast import literal_eval
from pipeline.utils import SourceMedia, Feature, Interval
from pipeline.utils import SourceMedia, Source, Feature, Interval

# for loudness detection
import soundfile
import pyloudnorm
@@ -114,7 +116,7 @@ class LaughterFeatureExtractor(FeatureExtractor):
for laugh in laughs:
start, end = laugh
self.features.append(Feature(interval=Interval(start=start, end=end),
source="laughter", path=file.path))
source=file, feature_extractor="laughter"))
# TODO: implement options eg minimum feature length

# adjust features
@@ -165,7 +167,12 @@ class RandomFeatureExtractor(FeatureExtractor):
duration = random.random() * self.MAX_DURATION
start = random.random() * file.duration() - duration
self.features.append(Feature(interval=Interval(start=start, duration=duration),
source="random", path=file.path))
source=file, feature_extractor="random"))

def teardown(self):
pass


class LoudAudioFeatureExtractor(FeatureExtractor):
"""Feature extractor for loud audio detection.



Loading…
取消
儲存