steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

BandwiseBinaryClassifier is work in progress; TODO: Shape Piping.

Browse Source
This commit is contained in:
Si11ium
2020-05-04 18:45:13 +02:00
parent e4f6506a4b
commit 451f78f820
7 changed files with 190 additions and 42 deletions

4
main_inference.py

@ -1,7 +1,7 @@
from torch.utils.data import DataLoader, Dataset
from torchvision.transforms import Compose, ToTensor
from ml_lib.audio_toolset.audio_io import Melspectogram, NormalizeLocal
from ml_lib.audio_toolset.audio_io import AudioToMel, NormalizeLocal
# Dataset and Dataloaders
# =============================================================================
@ -11,7 +11,7 @@ from ml_lib.utils.model_io import SavedLightningModels
from util.config import MConfig
from util.logging import MLogger
transforms = Compose([Melspectogram(), ToTensor(), NormalizeLocal()])
transforms = Compose([AudioToMel(), ToTensor(), NormalizeLocal()])
# Datasets
from datasets.binar_masks import BinaryMasksDataset