Skip to content

Commit

Permalink
Replace deprecated DiffPose module
Browse files Browse the repository at this point in the history
  • Loading branch information
eigenvivek committed Mar 11, 2024
1 parent b9298fe commit b3fc4d2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions experiments/deepfluoro/register.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@
import torch
from diffdrr.drr import DRR
from diffdrr.metrics import MultiscaleNormalizedCrossCorrelation2d
from diffdrr.pose import RigidTransform, convert
from torchvision.transforms.functional import resize
from tqdm import tqdm

from diffpose.calibration import RigidTransform, convert
from diffpose.deepfluoro import DeepFluoroDataset, Evaluator, Transforms
from diffpose.metrics import DoubleGeodesic, GeodesicSE3
from diffpose.registration import PoseRegressor, SparseRegistration
Expand Down Expand Up @@ -38,7 +38,7 @@ def __init__(

self.geodesics = GeodesicSE3()
self.doublegeo = DoubleGeodesic(sdr=self.specimen.focal_len / 2)
self.criterion = MultiscaleNormalizedCrossCorrelation2d([None, 13], [0.5, 0.5])
self.criterion = MultiscaleNormalizedCrossCorrelation2d([None, 9], [0.5, 0.5])
self.transforms = Transforms(self.drr.detector.height)
self.parameterization = parameterization
self.convention = convention
Expand Down

0 comments on commit b3fc4d2

Please sign in to comment.