MaHaWo commited on
Commit
3dc55b1
Β·
1 Parent(s): cf40c2c

rename new model to google_perch_lite

Browse files
{google_perch_tflite β†’ google_perch_lite}/default.yml RENAMED
File without changes
{google_perch_tflite β†’ google_perch_lite}/labels.txt RENAMED
File without changes
{google_perch_tflite β†’ google_perch_lite}/model.py RENAMED
@@ -32,7 +32,7 @@ class Model(ModelBase):
32
 
33
  # base class loads the model and labels
34
  super().__init__(
35
- "google_perch_tflite",
36
  model_path,
37
  labels_path,
38
  num_threads=num_threads,
 
32
 
33
  # base class loads the model and labels
34
  super().__init__(
35
+ "google_perch_lite",
36
  model_path,
37
  labels_path,
38
  num_threads=num_threads,
{google_perch_tflite β†’ google_perch_lite}/model.tflite RENAMED
File without changes
{google_perch_tflite β†’ google_perch_lite}/preprocessor.py RENAMED
@@ -1,24 +1,26 @@
1
- import numpy as np
2
  import iSparrow.preprocessor_base as ppb
3
 
4
 
5
  class Preprocessor(ppb.PreprocessorBase):
6
 
7
- def __init__(self,
 
8
  sample_rate: int = 32000,
9
  sample_secs: float = 5.0,
10
  resample_type: str = "kaiser_fast",
11
- **kwargs ):
 
12
 
13
  super().__init__(
14
- "google_perch_tflite",
15
  sample_rate=sample_rate,
16
  sample_secs=sample_secs,
17
  resample_type=resample_type,
18
  **kwargs
19
  )
20
 
21
- def process_audio_data(self, rawdata: np.array)->np.array:
22
 
23
  # raise when sampling rate is unequal.
24
  if self.actual_sampling_rate != self.sample_rate:
@@ -48,17 +50,16 @@ class Preprocessor(ppb.PreprocessorBase):
48
  split = temp
49
 
50
  self.chunks.append(split)
51
-
52
  print(
53
  "process audio data google: complete, read ",
54
  str(len(self.chunks)),
55
  "chunks.",
56
- flush=True
57
  )
58
 
59
  return self.chunks
60
 
61
-
62
  @classmethod
63
  def from_cfg(cls, cfg: dict):
64
 
 
1
+ import numpy as np
2
  import iSparrow.preprocessor_base as ppb
3
 
4
 
5
  class Preprocessor(ppb.PreprocessorBase):
6
 
7
+ def __init__(
8
+ self,
9
  sample_rate: int = 32000,
10
  sample_secs: float = 5.0,
11
  resample_type: str = "kaiser_fast",
12
+ **kwargs
13
+ ):
14
 
15
  super().__init__(
16
+ "google_perch_lite",
17
  sample_rate=sample_rate,
18
  sample_secs=sample_secs,
19
  resample_type=resample_type,
20
  **kwargs
21
  )
22
 
23
+ def process_audio_data(self, rawdata: np.array) -> np.array:
24
 
25
  # raise when sampling rate is unequal.
26
  if self.actual_sampling_rate != self.sample_rate:
 
50
  split = temp
51
 
52
  self.chunks.append(split)
53
+
54
  print(
55
  "process audio data google: complete, read ",
56
  str(len(self.chunks)),
57
  "chunks.",
58
+ flush=True,
59
  )
60
 
61
  return self.chunks
62
 
 
63
  @classmethod
64
  def from_cfg(cls, cfg: dict):
65