Would like to get rid of RotatePrepocessDataset
tonbo/data.py:class RotatePrepocessDataset(Dataset)
is used in scripts/run.py
and scripts/run2.py
. I think this can be removed. Conversion to polar is not used any more. Rotation should be done by torchvision.transforms.RandomRotation(degrees)
. We can just set degrees=[rot_angle, rot_angle]
, to rotate images by a specified angle.
Edited by Taro Kiritani