refactorings

This commit is contained in:
iperov 2019-04-24 11:28:58 +04:00
parent e1da9c56b4
commit 947feac047
11 changed files with 254 additions and 215 deletions

View file

@ -43,12 +43,11 @@ class SampleGeneratorImageTemporal(SampleGeneratorBase):
raise ValueError('No training data provided.')
mult_max = 4
l = samples_len - (self.temporal_image_count-1)*mult_max + 1
if l < 0:
samples_sub_len = samples_len - (self.temporal_image_count-1)*mult_max
if samples_sub_len <= 0:
raise ValueError('Not enough samples to fit temporal line.')
shuffle_idxs = []
samples_sub_len = samples_len - l + 1
while True: