MCPcopy Create free account

hub / github.com/akanazawa/cmr / types & classes

Types & classes30 in github.com/akanazawa/cmr

↓ 8 callersClassNeuralRenderer
This is the core pytorch function to call. Every torch NMR has a chainer NMR. Only fwd/bwd once per iteration.
nnutils/nmr.py:140
↓ 2 callersClassNMR
nnutils/nmr.py:29
↓ 2 callersClassRender
nnutils/nmr.py:100
↓ 2 callersClassUnsqueeze
nnutils/net_blocks.py:15
↓ 1 callersClassCodePredictor
nnutils/mesh_net.py:205
↓ 1 callersClassEncoder
Current: Resnet with 4 blocks (x32 spatial dim reduction) Another conv with stride 2 (x64) This is sent to 2 fc layers with final out
nnutils/mesh_net.py:65
↓ 1 callersClassFlatten
nnutils/net_blocks.py:11
↓ 1 callersClassLaplacian
nnutils/laplacian.py:33
↓ 1 callersClassPerceptualLoss
nnutils/perceptual_loss.py:14
↓ 1 callersClassQuatPredictor
nnutils/mesh_net.py:160
↓ 1 callersClassResNetConv
nnutils/mesh_net.py:41
↓ 1 callersClassScalePredictor
nnutils/mesh_net.py:175
↓ 1 callersClassShapePredictor
Outputs mesh deformations
nnutils/mesh_net.py:138
↓ 1 callersClassShapeTester
benchmark/evaluate.py:29
↓ 1 callersClassShapeTrainer
experiments/shape.py:55
↓ 1 callersClassTeapotModel
nnutils/nmr.py:234
↓ 1 callersClassTexturePredictorUV
Outputs mesh texture
nnutils/mesh_net.py:92
↓ 1 callersClassTransPredictor
Outputs [tx, ty] or [tx, ty, tz]
nnutils/mesh_net.py:187
↓ 1 callersClassVisualizer
utils/visualizer.py:9
ClassArcball
Virtual Trackball Control. >>> ball = Arcball() >>> ball = Arcball(initial=numpy.identity(4)) >>> ball.place([320, 320], 320) >>> bal
utils/transformations.py:1515
ClassBaseDataset
img, mask, kp, pose data loader
data/base.py:48
ClassCUBDataset
CUB Data loader
data/cub.py:45
ClassEdgeLoss
Edge length should not diverge from the original edge length. On initialization computes the current edge lengths.
nnutils/loss_utils.py:234
ClassLaplacianLoss
Encourages minimal mean curvature shapes.
nnutils/loss_utils.py:309
ClassMeshNet
nnutils/mesh_net.py:222
ClassMeshPredictor
nnutils/predictor.py:31
ClassPerceptualTextureLoss
nnutils/loss_utils.py:349
ClassTester
nnutils/test_utils.py:64
ClassTrainer
nnutils/train_utils.py:56
ClassVisRenderer
Utility to render meshes using pytorch NMR faces are F x 3 or 1 x F x 3 numpy
utils/bird_vis.py:18