Skip to content
Snippets Groups Projects

Resolve "Image creation bugs with 0 size windows"

Closed Raphael Sturgis requested to merge 21-image-creation-bugs-with-0-size-windows into main
3 files
+ 129
12
Compare changes
  • Side-by-side
  • Inline
Files
3
import tqdm as tqdm
import tqdm as tqdm
 
from skais.process.data_augmentation.data_transformer import DataTransformer
from skais.process.data_augmentation.flip import Flip
from skais.process.data_augmentation.flip import Flip
from skais.process.data_augmentation.pipeline import Pipeline
from skais.process.data_augmentation.pipeline import Pipeline
from skais.process.data_augmentation.translator import Translator
from skais.process.data_augmentation.translator import Translator
class AugmentationEngine:
class AugmentationEngine:
def __init__(self, translation_values, flip_values):
def __init__(self, translation_values=None, flip_values=None, keep_original=True):
self.pipelines = []
self.pipelines = []
 
if keep_original:
 
self.pipelines.append(DataTransformer())
for tv_long, tv_lat in translation_values:
if translation_values is not None:
self.pipelines.append(Pipeline([Translator(tv_long, tv_lat)]))
for tv_long, tv_lat in translation_values:
 
self.pipelines.append(Pipeline([Translator(tv_long, tv_lat)]))
for fv_meridian, fv_parallel in flip_values:
if flip_values is not None:
self.pipelines.append(Pipeline([Flip(fv_meridian, fv_parallel)]))
for tv_long, tv_lat in translation_values:
translator = Translator(tv_long, tv_lat)
for fv_meridian, fv_parallel in flip_values:
for fv_meridian, fv_parallel in flip_values:
flip = Flip(fv_meridian, fv_parallel)
self.pipelines.append(Pipeline([Flip(fv_meridian, fv_parallel)]))
self.pipelines.append(Pipeline([translator, flip]))
 
if flip_values is not None and translation_values is not None:
 
for tv_long, tv_lat in translation_values:
 
translator = Translator(tv_long, tv_lat)
 
for fv_meridian, fv_parallel in flip_values:
 
flip = Flip(fv_meridian, fv_parallel)
 
self.pipelines.append(Pipeline([translator, flip]))
def transform(self, x, verbose=0):
def transform(self, x, verbose=0):
results = x.copy()
results = x.copy()
Loading