Skip to content

fixes in rotate and reflect functions #12

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
cwsmith merged 5 commits into from
Jul 2, 2025
Merged

fixes in rotate and reflect functions #12

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
7c1df42
broken plotting of rotated frames
cwsmith Jun 10, 2025
4bbf134
removed bug where train_loss and val_loss are reset to empty lists af…
sridhs21 Jun 16, 2025
6fbc772
fixed bugs in rotate and reflect functions
sridhs21 Jun 25, 2025
fc8b2bb
removed commented lines of code in rotate and reflect function
sridhs21 Jun 27, 2025
bfb0c98
removed plotSimple
sridhs21 Jun 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 10 additions & 8 deletions XPointMLTest.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,9 +63,12 @@ def rotate(frameData,deg):
if deg not in [90, 180, 270]:
print(f"invalid rotation specified... exiting")
sys.exit()

psi = v2.functional.rotate(frameData["psi"], deg, v2.InterpolationMode.BILINEAR)
all = v2.functional.rotate(frameData["all"], deg, v2.InterpolationMode.BILINEAR)
mask = v2.functional.rotate(frameData["mask"], deg, v2.InterpolationMode.BILINEAR)
# For mask, use nearest neighbor interpolation to preserve binary values
mask = v2.functional.rotate(frameData["mask"], deg, v2.InterpolationMode.NEAREST)

return {
"fnum": frameData["fnum"],
"rotation": deg,
Expand All @@ -83,9 +86,11 @@ def reflect(frameData,axis):
if axis not in [0,1]:
print(f"invalid reflection axis specified... exiting")
sys.exit()
psi = torch.flip(frameData["psi"][0], dims=(axis,)).unsqueeze(0)
all = torch.flip(frameData["all"], dims=(axis,))
mask = torch.flip(frameData["mask"][0], dims=(axis,)).unsqueeze(0)

psi = torch.flip(frameData["psi"], dims=(axis+1,))
all = torch.flip(frameData["all"], dims=(axis+1,))
mask = torch.flip(frameData["mask"], dims=(axis+1,))

return {
"fnum": frameData["fnum"],
"rotation": 0,
Expand Down Expand Up @@ -841,9 +846,6 @@ def main():
t2 = timer()
print("time (s) to prepare model: " + str(t2-t1))

train_loss = []
val_loss = []

num_epochs = args.epochs
for epoch in range(start_epoch, num_epochs):
train_loss.append(train_one_epoch(model, train_loader, criterion, optimizer, device))
Expand Down Expand Up @@ -939,4 +941,4 @@ def main():
print("total time (s): " + str(t5-t0))

if __name__ == "__main__":
main()
main()