I changed few if else statement in rotation.py

HealthML / self-supervised-3d-tasks

Apache License 2.0

185 stars 39 forks source link

I changed few if else statement in rotation.py #11

Closed ramchandracheke closed 3 years ago

ramchandracheke commented 3 years ago

Hey there,

Thanks for your amazing code. I think there are minor changes in preprocessing ... rotation.py file for defining 10 possible rotation, Please have a look.

Kind regards, Ram

aihamtaleb commented 3 years ago

Thanks for your proposed changes, they certainly improve the rotation preprocessing.