def _test_bg_subtraction2(self): p = Preprocessor(10) s = p.load_npy('./test.npy') generator = DataPreparator("", "", 512) samples1=len(s[0,:]) snew,sbg = generator.bg_subtraction(s) samples2=len(snew[0,:]) self.assertGreater(samples1, samples2)
def _test_bg_subtraction(self): p=Preprocessor(10) s=p.load_npy('./test.npy') generator = DataPreparator("", "", 512) mask, vector = generator.create_mask(s) self.assertGreater(np.sum(vector), 120)