Image-and-3D-Model-Creator

Build error

App Files Files Community

Image-and-3D-Model-Creator / PIFu /lib /model /DepthNormalizer.py

cownclown

F4RF4R4

Duplicate from F4RF4R4/Image-and-3D-Model-Creator

2d1920d over 3 years ago

Raw History Blame Contribute Delete

497 Bytes

import torch
import torch.nn as nn
import torch.nn.functional as F


class DepthNormalizer(nn.Module):
    def __init__(self, opt):
        super(DepthNormalizer, self).__init__()
        self.opt = opt

    def forward(self, z, calibs=None, index_feat=None):
        '''
        Normalize z_feature
        :param z_feat: [B, 1, N] depth value for z in the image coordinate system
        :return:
        '''
        z_feat = z * (self.opt.loadSize // 2) / self.opt.z_size
        return z_feat

	import torch
	import torch.nn as nn
	import torch.nn.functional as F


	class DepthNormalizer(nn.Module):
	def __init__(self, opt):
	super(DepthNormalizer, self).__init__()
	self.opt = opt

	def forward(self, z, calibs=None, index_feat=None):
	'''
	Normalize z_feature
	:param z_feat: [B, 1, N] depth value for z in the image coordinate system
	:return:
	'''
	z_feat = z * (self.opt.loadSize // 2) / self.opt.z_size
	return z_feat