MCPcopy Create free account

hub / github.com/vt-vl-lab/3d-photo-inpainting / functions

Functions141 in github.com/vt-vl-lab/3d-photo-inpainting

⨍ Functions 141◇ Types & classes 13
↓ 27 callersMethodcat
(self, A, B)
networks.py:362
↓ 13 callersFunctionout_fmt
(input, cur_id_b, cur_id_self, cur_id_a, ply_flag)
mesh.py:200
↓ 12 callersFunctionspectral_norm
(module, mode=True)
networks.py:497
↓ 8 callersFunctionextrapolate
(global_mesh, info_on_pix, image, depth, other
mesh_tools.py:81
↓ 8 callersMethodforward_3P
(self, mask, context, rgb, disp, edge, unit_length=128, cuda=None)
networks.py:301
↓ 8 callersFunctiongroup_edges
(1) add_new_node(G, node) : add "node" to graph "G" (2) add_new_edge(G, node_a, node_b) : add edge "node_a--node_b" to graph "G" (3) exce
mesh.py:379
↓ 8 callersFunctionupdate_info
(mapping_dict, info_on_pix, *meshes)
mesh_tools.py:792
↓ 7 callersMethod__init__
(self, in_ch, out_ch, bn=True, sample='none-3', activ='relu', conv_bias=False)
networks.py:107
↓ 7 callersMethodload
Load model from file. Args: path (str): file path
MiDaS/monodepth_net.py:82
↓ 7 callersMethodupsample
(self, feat, mask)
networks.py:365
↓ 6 callersFunctioncrop_maps_by_size
(size, *imaps)
mesh_tools.py:1068
↓ 6 callersFunctiondilate_valid_size
(isize_dict, imap, dilate=[0, 0])
mesh_tools.py:450
↓ 6 callersFunctionget_valid_size
(imap)
mesh_tools.py:441
↓ 6 callersFunctionupdate_status
(2) clear_node_feat(G, *fts) : Clear all the node feature on graph G. (6) get_cross_nes(x, y) : Get the four cross neighbors of pixel(x, y).
mesh.py:330
↓ 5 callersFunctionconvert2tensor
(input_dict)
mesh_tools.py:1075
↓ 5 callersFunctionrequire_depth_edge
(context_edge, mask)
utils.py:1057
↓ 4 callersFunctionrolling_window
(a, window, strides)
bilateral_filtering.py:204
↓ 4 callersMethodview_changed
(self)
mesh.py:2153
↓ 3 callersMethod__init__
Init. Args: scale_factor (float): scaling mode (str): interpolation mode
MiDaS/monodepth_net.py:97
↓ 3 callersFunctioncombine_end_node
(mesh, edge_mesh, edge_ccs, depth)
mesh.py:536
↓ 3 callersFunctionfilter_irrelevant_edge_new
(self_edge, comp_edge, other_edges, other_edges_with_id, current_edge_id, context, depth, mesh, context_cc, sp
utils.py:66
↓ 3 callersFunctionget_edge_from_nodes
(context_cc, erode_context_cc, mask_cc, edge_cc, extend_edge_cc, H, W, mesh)
mesh_tools.py:947
↓ 3 callersFunctionget_map_from_ccs
(ccs, height, width, condition_input=None, condition=None, real_id=False, id_shift=0)
mesh_tools.py:875
↓ 3 callersMethodreinit_camera
(self, fov)
mesh.py:2162
↓ 3 callersMethodrotate
(self, axis=[1,0,0], angle=0)
mesh.py:2150
↓ 3 callersFunctionsize_operation
(size_a, size_b, operation)
mesh_tools.py:459
↓ 3 callersMethodtranslate
(self, trans=[0,0,0])
mesh.py:2147
↓ 2 callersFunctionDL_inpaint_edge
(mesh, info_on_pix, config, image,
mesh.py:1387
↓ 2 callersMethodadd_border
(self, input, mask_flag, PCONV=True)
networks.py:158
↓ 2 callersFunctionclean_far_edge_new
(input_edge, end_depth_maps, mask, context, global_mesh, info_on_pix, self_edge, inpaint_id, config)
utils.py:114
↓ 2 callersFunctionclean_folder
(folder, img_exts=['.png', '.jpg', '.npy'])
boostmonodepth_utils.py:47
↓ 2 callersFunctioncontext_and_holes
(mesh, edge_ccs, config, specific_edge_id, specific_edge_loc, depth_feat_model, connect_
mesh.py:904
↓ 2 callersFunctiondepth_inpainting
(context_cc, extend_context_cc, erode_context_cc, mask_cc, mesh, config, union_size, depth_feat_model, edge_ou
mesh_tools.py:751
↓ 2 callersFunctionfilter_edge
(mesh, edge_ccs, config, invalid=False)
mesh_tools.py:47
↓ 2 callersMethodinit_weights
initialize network's weights init_type: normal | xavier | kaiming | orthogonal https://github.com/junyanz/pytorch-CycleGAN-an
networks.py:12
↓ 2 callersFunctionmax_rectangle_size
(histogram)
utils.py:1365
↓ 2 callersFunctionopen_small_mask
(mask, context, open_iteration, kernel)
utils.py:54
↓ 2 callersFunctionplan_path
(mesh, info_on_pix, cc, end_pt, global_mesh, input_edge, mask, valid_map, inpaint_id, npath_map=None, fpath_ma
utils.py:438
↓ 2 callersFunctionrecursive_add_edge
(edge_mesh, mesh, info_on_pix, cur_node, mark)
mesh_tools.py:815
↓ 2 callersFunctionrefine_depth_around_edge
(mask_depth, far_edge, uncleaned_far_edge, near_edge, mask, all_depth, config)
utils.py:1224
↓ 2 callersFunctionrefresh_node
(old_node, old_feat, new_node, new_feat, mesh, stime=False)
utils.py:550
↓ 2 callersFunctionremove_redundant_edge
(mesh, edge_mesh, edge_ccs, info_on_pix, config, redundant_number=1000, invalid=False, spdb=False)
mesh.py:630
↓ 2 callersMethodrender
(self)
mesh.py:2156
↓ 1 callersFunctionacc_shape
(i)
bilateral_filtering.py:208
↓ 1 callersMethodadd_border
(self, input, channel_pad_1=None)
networks.py:285
↓ 1 callersFunctionbilateral_filter
(depth, config, discontinuity_map=None, HR=False, mask=None, window_size=False)
bilateral_filtering.py:105
↓ 1 callersFunctionbuild_connection
(mesh, cur_node, dst_node)
mesh_tools.py:801
↓ 1 callersFunctioncalculate_fov
(mesh)
mesh.py:106
↓ 1 callersFunctionclear_node_feat
(G, fts)
mesh.py:340
↓ 1 callersFunctioncreate_mesh
(depth, image, int_mtx, config)
mesh.py:28
↓ 1 callersFunctioncreate_placeholder
(context, mask, depth, fpath_map, npath_map, mesh, inpaint_id, edge_ccs, extend_edge_cc, all_edge_maps, self_e
utils.py:586
↓ 1 callersFunctionenlarge_border
(mesh, info_on_pix, depth, image, config)
mesh_tools.py:504
↓ 1 callersFunctionfill_dummy_bord
(mesh, info_on_pix, image, depth, config)
mesh_tools.py:471
↓ 1 callersFunctionfill_missing_node
(mesh, info_on_pix, image, depth)
mesh_tools.py:519
↓ 1 callersFunctionfind_anchors
(matrix)
utils.py:1393
↓ 1 callersMethodforward
(self, input_feat, refine_border=False, sample=False, PCONV=True)
networks.py:195
↓ 1 callersMethodforward
(self, x, refine_border=False)
networks.py:316
↓ 1 callersMethodforward
(self, input, add_border=False)
networks.py:388
↓ 1 callersMethodforward
Forward pass. Args: x (tensor): input Returns: tensor: interpolated data
MiDaS/monodepth_net.py:110
↓ 1 callersFunctiongenerate_face
(mesh, info_on_pix, config)
mesh.py:195
↓ 1 callersFunctiongenerate_init_node
(mesh, config, min_node_in_cc)
mesh.py:158
↓ 1 callersFunctionget_MiDaS_samples
(image_folder, depth_folder, config, specific=None, aft_certain=None)
utils.py:838
↓ 1 callersFunctionget_depth_from_maps
(context_map, mask_map, depth_map, H, W, log_depth=False)
mesh_tools.py:992
↓ 1 callersFunctionget_depth_from_nodes
(context_cc, erode_context_cc, mask_cc, H, W, mesh, log_depth=False)
mesh_tools.py:1013
↓ 1 callersFunctionget_neighbors
(mesh, node)
mesh.py:192
↓ 1 callersFunctionget_rgb_from_nodes
(context_cc, erode_context_cc, mask_cc, H, W, mesh)
mesh_tools.py:1049
↓ 1 callersFunctionmax_size
(mat, value=0)
utils.py:1354
↓ 1 callersFunctionoutput_3d_photo
(verts, colors, faces, Height, Width, hFov, vFov, tgt_poses, video_traj_types, ref_pose, o
mesh.py:2167
↓ 1 callersFunctionpath_planning
(num_frames, x, y, z, path_type='')
utils.py:29
↓ 1 callersFunctionplan_path_e2e
(mesh, cc, end_pts, global_mesh, input_edge, mask, valid_map, inpaint_id, npath_map=None, fpath_map=None)
utils.py:341
↓ 1 callersFunctionread_MiDaS_depth
(disp_fi, disp_rescale=10., h=None, w=None)
utils.py:942
↓ 1 callersFunctionread_ply
(mesh_fi)
mesh.py:2069
↓ 1 callersFunctionreassign_floating_island
(mesh, info_on_pix, image, depth)
mesh.py:238
↓ 1 callersFunctionrefine_color_around_edge
(mesh, info_on_pix, edge_ccs, config, spdb=False)
utils.py:1064
↓ 1 callersFunctionrefresh_bord_depth
(mesh, info_on_pix, image, depth)
mesh_tools.py:547
↓ 1 callersMethodreinit_mesh
(self, verts, faces, colors)
mesh.py:2159
↓ 1 callersFunctionrelabel_node
(mesh, nodes, cur_node, new_node)
mesh_tools.py:35
↓ 1 callersFunctionremove_dangling
(mesh, edge_ccs, edge_mesh, info_on_pix, image, depth, config)
mesh.py:733
↓ 1 callersFunctionremove_node_feat
(mesh, *feats)
mesh.py:322
↓ 1 callersFunctionreproject_3d_int_detail
(sx, sy, z, k_00, k_02, k_11, k_12, w_offset, h_offset)
mesh.py:129
↓ 1 callersFunctionresize_depth
Resize numpy (or image read by imageio) depth map Args: depth (numpy): depth width (int): image width height (int): image
boostmonodepth_utils.py:56
↓ 1 callersFunctionrun_boostmonodepth
(img_names, src_folder, depth_folder)
boostmonodepth_utils.py:13
↓ 1 callersFunctionrun_depth
Run MonoDepthNN to compute depth maps. Args: input_path (str): path to input folder output_path (str): path to output folder
MiDaS/run.py:14
↓ 1 callersFunctionsmooth_cntsyn_gap
(init_depth_map, mask_region, context_region, init_mask_region=None)
utils.py:913
↓ 1 callersFunctionsparse_bilateral_filtering
config: - filter_size
bilateral_filtering.py:4
↓ 1 callersFunctiontear_edges
(mesh, threshold = 0.00025, xy2depth=None)
mesh.py:65
↓ 1 callersFunctionvis_depth_discontinuity
config: -
bilateral_filtering.py:48
↓ 1 callersFunctionweights_init
(init_type='gaussian')
networks.py:40
↓ 1 callersFunctionwrite_depth
Write depth map to pfm and png file. Args: path (str): filepath without extension depth (array): depth
MiDaS/MiDaS_utils.py:168
↓ 1 callersFunctionwrite_ply
(image, depth, int_mtx, ply_name, config,
mesh.py:1820
Method__init__
(self)
networks.py:9
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bia
networks.py:63
Method__init__
(self, layer_size=7, upsampling_mode='nearest')
networks.py:135
Method__init__
(self, residual_blocks=8, init_weights=True)
networks.py:238
Method__init__
(self, layer_size=7, upsampling_mode='nearest', add_hole_mask=False, add_two_layer=False, add_border=False)
networks.py:333
Method__init__
(self, use_sigmoid=True, use_spectral_norm=True, init_weights=True, in_channels=None)
networks.py:431
Method__init__
(self, dim, dilation=1)
networks.py:475
Method__init__
(self, fov, verts, faces, colors,
mesh.py:2124
Method__init__
Init. Args: path (str, optional): Path to saved model. Defaults to None. features (int, optional): Number of features
MiDaS/monodepth_net.py:14
Method__init__
Init. Args: features (int): number of features
MiDaS/monodepth_net.py:128
next →1–100 of 141, ranked by callers