LumenPallidium / neural-file-sorter

A neural network based file sorter. Trains an autoencoder to sort images or audio based on the similarity of their encodings, or uses the OpenAI CLIP model.
MIT License
27 stars 1 forks source link

add data augementation and transformation options #6

Open LumenPallidium opened 3 years ago

LumenPallidium commented 3 years ago

currently only scaling exists, i'd like to have options like crop, crop and scale, as well as affine transformations on images

LumenPallidium commented 3 years ago

added affine transformations in 23c671a

LumenPallidium commented 3 years ago

changed scaling to random scale and random crop in 0d73e3d

LumenPallidium commented 3 years ago

added a scale only transformation in 6bc5afdc07ed84adffcf0b5fd1d7ef1c12ecdb7f for the final embedding step as well as data viewing