init project

app.py

@@ -36,11 +36,14 @@ import matplotlib.pyplot as pl
 
 from modules.mobilesamv2.utils.transforms import ResizeLongestSide
 from modules.pe3r.models import Models
+import torchvision.transforms as tvf
+
 
 silent = True
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 pe3r = Models(device)
 
+
 def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
                                  cam_color=None, as_pointcloud=False,
                                  transparent_cams=False, silent=False):
@@ -474,7 +477,6 @@ def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, ni
     loss = scene_1.compute_global_alignment(tune_flg=True, init='mst', niter=niter, schedule=schedule, lr=lr)
 
     try:
-        import torchvision.transforms as tvf
         ImgNorm = tvf.Compose([tvf.ToTensor(), tvf.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
         for i in range(len(imgs)):
             # print(imgs[i]['img'].shape, scene.imgs[i].shape, ImgNorm(scene.imgs[i])[None])