def test_preprocess(path, training):
prep.preprocess(TEST_TRAILER, train=training, n_subclips=2)
name = os.listdir(path)[0]
digit = name[-1]
assert len(os.listdir(path)) == 2
assert 13 <= len(os.listdir(f'{path}/{name}/{digit}')) <= 18
shutil.rmtree(path)
def predict_single(self, movie_name, verbose=False):
print('----------GETTING PREDICTION-----------')
prep.preprocess(movie_name, train=False, n_subclips=1, verbose=verbose)
name = movie_name.rsplit('.', 1)[0] + '_0' #Getting appropriate name
image, _ = load_data([name],
train=False,
verbose=verbose,
batch_size=1)
prediction = self.forward(image).detach().unsqueeze(-1).item()
print('---------------------------------------')
return prediction
def train_prep():
yield prep.preprocess('86.Малыш на драйве.mp4', train=True, verbose=True)
shutil.rmtree(TRAINING_PATH)
def test_prep():
yield prep.preprocess('85.Дикая.mp4', train=False, verbose=True)
shutil.rmtree(PREDICTION_PATH)