salehahr / graph-extraction-networks

Extraction of graphs from skeletonised images.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

correct data augmentation for adjacency matrices

reoux opened this issue · comments

commented

Notes

Original

image

Augmented

Transformations from top left: flip_horz, flip_vert + flip_horz, orig, orig

skeletonised adj_matr
image image
commented

In progress, s. ee88d7d.

Notes

6640B5B9-FF87-47D1-A38C-5F6878F81F08

commented

I think everything is working as it should, as of 642f44f.

Checking the augmentation using TestThirdNetworkDataAugmentation.test_data_aug in tests/test_data_augmentation.py:

  1 2 3
Before image image image
After image image image
  Vertical flip Horizontal flip Both flips

Before augmentation

image

After augmentation

image