Fixes in augment() and in validation dataset generation

juglab / cryoCARE_pip

PIP package of cryoCARE

BSD 3-Clause "New" or "Revised" License

25 stars 14 forks source link

Fixes in augment() and in validation dataset generation #34

Closed rdrighetto closed 1 year ago

rdrighetto commented 1 year ago

cryoCARE_extract_train_data.py now runs until completion with the here proposed changes (please check if they make sense). I'm now doing a full run to see if the quality issues are fixed.

tibuch commented 1 year ago

Oh, I see. Is your sample_shape the same in all dimensions?

We have to be careful to only rotate around the tilt-axis.

rdrighetto commented 1 year ago

Just to answer here: yes, in our case sample_shape is always cubic and tilt axis is always Y, but I understand the code needs to be robust to deal with cases that differ from that. Thanks for the new fix -- I will report on that soon!