Stage 1 - apply_warping_field - vs seems to be wrong size

johndpope / MegaPortrait-hack

Using Claude Opus to reverse engineer code from MegaPortraits: One-shot Megapixel Neural Head Avatars

https://arxiv.org/abs/2207.07621

42 stars 7 forks source link

Closed johndpope closed 1 month ago

johndpope commented 1 month ago

vs shape: torch.Size([1, 96, 16, 32, 32]) should be [1, 96, 16, 64, 64]

@kwentar - Aleksey I add the extra ResBlock3d_96

I think there's something wrong with the avgpool

TODO - inspect any deviation in the ResBlock_Custom vs ResBlock3D_Adaptive / there totally built separately....

johndpope commented 1 month ago

for now I rip out one of the avgpool layers - then we get to 64x64 - otherwise its 32x32. on to g2d

johndpope commented 1 month ago

G2d is ok - but have to fix eyeglaze loss