direct training

2023-05-23 22:57:08 +02:00
parent 843c9ac4b0
commit 53217c6e5d
8 changed files with 470 additions and 4 deletions
@@ -43,6 +43,7 @@ class UV2Mesh(nn.Module):
        if uv.shape[1] == 3 or uv.shape[1] == 7:
            uv = torch.permute(uv, [0,2,3,1])
        if uv.shape[1] != 512:
+            raise NotImplementedError
            uv = Resize(uv[:,:,None], size=(512, 256), interpolation=InterpolationMode.BILINEAR)
            
        # batch_size = uv.shape[0]