Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AutodeskAILab/BRepNet
/ functions
Functions
409 in github.com/AutodeskAILab/BRepNet
⨍
Functions
409
◇
Types & classes
24
↓ 1 callers
Method
save_logits
Save logits for this batch
models/brepnet.py:863
↓ 1 callers
Function
save_results
(log_dir, opts, results)
train/train.py:11
↓ 1 callers
Method
scale_from_point_grids
(self, grids)
pipeline/extract_brepnet_data_from_step.py:643
↓ 1 callers
Method
sort_faces_by_num_coedges
Sort the faces in the batch data by the number of coedges on each face. Apply the permutation to the entirety of the da
dataloaders/brepnet_dataset_old.py:602
↓ 1 callers
Method
sphere_feature
(self, face)
pipeline/extract_brepnet_data_from_step.py:220
↓ 1 callers
Method
standardize_brep_feature_data
Standardize brep feature data
dataloaders/brepnet_dataset_old.py:469
↓ 1 callers
Method
straight_edge_feature
(self, edge)
pipeline/extract_brepnet_data_from_step.py:350
↓ 1 callers
Method
torus_feature
(self, face)
pipeline/extract_brepnet_data_from_step.py:227
↓ 1 callers
Method
validate
Validate that the faces in the given STEP file map with the face indices as defined by the OBJ meshes extracted with the dataset.
pipeline/face_index_validator.py:57
↓ 1 callers
Method
variance
(self)
pipeline/running_stats.py:57
↓ 1 callers
Function
view_file
(obj_file, seg_file)
visualization/view_segmentation.py:6
↓ 1 callers
Method
weights_init
(self, m)
models/uvnet_encoders.py:87
↓ 1 callers
Method
weights_init
(self, m)
models/uvnet_encoders.py:133
↓ 1 callers
Function
write_col_headings
Write the headings for the worksheet columns
train/grid_search.py:13
↓ 1 callers
Function
write_results_to_excel
Write the results to an excel workbook
train/grid_search.py:45
↓ 1 callers
Function
write_results_to_workbook
Write out the data to a worksheet
train/grid_search.py:30
Method
__getitem__
Load a batch of data. Use a cache if its there. Cache the binary data if not.
dataloaders/brepnet_dataset_old.py:52
Method
__getitem__
Load the data for a body. Use a cache if its there. Cache the binary data if not.
dataloaders/brepnet_dataset.py:200
Method
__init__
(self, solid)
visualization/save_images_of_similar_solids.py:5
Method
__init__
(self, *args, **kwargs)
visualization/jupyter_segmentation_viewer.py:39
Method
__init__
(self, file_stem, step_folder, seg_folder=None, logit_folder=None)
visualization/jupyter_segmentation_viewer.py:70
Method
__init__
(self, data_source, max_num_faces_per_batch)
dataloaders/max_num_faces_sampler.py:6
Method
__init__
Construct the dataset from the given options
dataloaders/brepnet_dataset_old.py:14
Method
__init__
(self, opts, train_val_or_test)
dataloaders/brepnet_dataset.py:172
Method
__init__
This is the 2D convolutional network that extracts features from the B-rep face geometry described as 2D UV-grids (see Section 3.2, C
models/uvnet_encoders.py:106
Method
__init__
Initialize the layer
models/brepnet.py:171
Method
__init__
Initialization of a general BRepNet layer. num_mlp_layers - Number of layers we want in the MLP input_size - This
models/brepnet.py:222
Method
__init__
Initialization of the BRepNet output layer. num_mlp_layers - Number of layers we want in the MLP input_size - This
models/brepnet.py:301
Method
__init__
Initialize the class with the pathnames for the step file and corresponding segmentation file
pipeline/segmentation_file_crosschecker.py:13
Method
__init__
(self, step_file, mesh_dir)
pipeline/face_index_validator.py:53
Method
__init__
(self, topology, features, feature_schema)
pipeline/extract_brepnet_data_from_json.py:11
Method
__init__
(self)
pipeline/running_stats.py:10
Method
__init__
(self, step_file, output_dir, feature_schema, scale_body=True)
pipeline/extract_brepnet_data_from_step.py:48
Method
__init__
Create a mapper object for this list of bodies
pipeline/entity_mapper.py:22
Method
__iter__
Get an iterator over the batches
dataloaders/max_num_faces_sampler.py:17
Method
__len__
Get the number of batches
dataloaders/max_num_faces_sampler.py:11
Method
__len__
The dataloader returns the dataset already divided into mini-batches.
dataloaders/brepnet_dataset_old.py:45
Method
__len__
Get the number of bodies in the dataset
dataloaders/brepnet_dataset.py:193
Method
body_index
Find the index of a body
pipeline/entity_mapper.py:78
Method
clear
(self)
pipeline/running_stats.py:20
Method
click
called whenever a shape or edge is clicked
visualization/jupyter_segmentation_viewer.py:42
Method
configure_optimizers
(self)
models/brepnet.py:999
Method
display_faces_with_heatmap
(self, values, interval=None)
visualization/jupyter_segmentation_viewer.py:217
Function
evaluate_folder
( step_folder, feature_standardization, model=None, input_features=None
eval/evaluate_folder.py:152
Function
find_box
(solid)
utils/scale_utils.py:16
Function
find_checkpoint
(logs_dir, month_day, hour_min_second)
utils/checkpoint_utils.py:3
Method
forward
(self, x)
models/uvnet_encoders.py:93
Method
forward
(self, x)
models/uvnet_encoders.py:139
Method
forward
Forward pass through the MLP
models/brepnet.py:211
Method
forward
This layer performs the following steps 1) The matrix Psi is by permuting and concatenating the feature vecto
models/brepnet.py:249
Method
forward
This layer performs the following steps 1) The matrix Psi is by permuting and concatenating the feature vecto
models/brepnet.py:327
Method
forward
A forward pass through the network.
models/brepnet.py:566
Method
get_nr_of_edges
(self)
pipeline/entity_mapper.py:72
Method
get_nr_of_surfaces
(self)
pipeline/entity_mapper.py:75
Method
highlight_faces_with_indices
(self, indices)
visualization/jupyter_segmentation_viewer.py:200
Function
load_json_data
Load data from a json file
utils/data_utils.py:4
Function
load_labels
(label_pathname)
utils/data_utils.py:60
Function
load_npz_data
(npz_file)
utils/data_utils.py:41
Method
loop_index
Find the index of a loop
pipeline/entity_mapper.py:106
Method
primary_face_orientation
(self, face)
pipeline/entity_mapper.py:142
Method
print_parameter_info
Print more detailed info about the network parameters.
models/brepnet.py:587
Function
run_worker
(worker_args)
pipeline/extract_brepnet_data_from_json.py:139
Function
run_worker
(worker_args)
pipeline/extract_brepnet_data_from_step.py:896
Method
save_image_of_selected_faces
( self, pathname, values_for_faces, threshold )
visualization/save_images_of_similar_solids.py:14
Function
save_json_data
Export a data to a json file
utils/data_utils.py:10
Function
save_npz_data_without_uvnet_features
(output_pathname, data)
utils/data_utils.py:15
Function
scale_solid_to_unit_box
(solid)
utils/scale_utils.py:23
Method
select_face_callback
Callback from the notebook when we select a face
visualization/jupyter_segmentation_viewer.py:128
Method
shell_index
Find the index of a shell
pipeline/entity_mapper.py:92
Method
solid_index
Find the index of a solid
pipeline/entity_mapper.py:85
Method
test_block
(self)
tests/test_extract_brepnet_data_from_step.py:658
Method
test_block_fillet1
(self)
tests/test_extract_brepnet_data_from_step.py:684
Method
test_block_fillet3
(self)
tests/test_extract_brepnet_data_from_step.py:715
Method
test_block_hole
(self)
tests/test_extract_brepnet_data_from_step.py:743
Method
test_block_through_hole
(self)
tests/test_extract_brepnet_data_from_step.py:770
Method
test_block_torus_fillet3
(self)
tests/test_extract_brepnet_data_from_step.py:797
Method
test_brepnet_collate_fn
(self)
tests/test_dataloader.py:849
Method
test_build_coedges_of_edges_tensor
Test for the BRepNetDataset.build_coedges_of_edges_tensor() function
tests/test_dataloader.py:472
Method
test_build_coedges_of_faces_tensor
Test for the BRepNetDataset.build_coedges_of_faces_tensor() function
tests/test_dataloader.py:495
Method
test_cylinder
(self)
tests/test_extract_brepnet_data_from_step.py:830
Method
test_dataloader
(self)
models/brepnet.py:972
Method
test_dataloaders_equivalent
(self)
tests/test_dataloaders_equivalent.py:268
Method
test_epoch_end
Collate the results from all test batches
models/brepnet.py:824
Method
test_face_3_loops
(self)
tests/test_extract_brepnet_data_from_step.py:855
Method
test_feature_data
(self)
tests/test_extract_brepnet_data_from_step.py:981
Method
test_find_face_permutation
Test for the BRepNetDataset.find_face_permutation() function.
tests/test_dataloader.py:538
Method
test_find_inverse_permutation_np
(self)
tests/test_dataloader.py:600
Method
test_find_inverse_permutation_torch
(self)
tests/test_dataloader.py:609
Method
test_four_cone
(self)
tests/test_extract_brepnet_data_from_step.py:868
Method
test_full_dataloader
(self)
tests/test_dataloader.py:84
Method
test_full_dataset
(self)
tests/test_dataloaders_equivalent.py:280
Method
test_sphere
(self)
tests/test_extract_brepnet_data_from_step.py:877
Method
test_standardize_features
(self)
tests/test_dataloader.py:569
Method
test_step
Test on one batch
models/brepnet.py:816
Method
test_three_concave_edge
(self)
tests/test_extract_brepnet_data_from_step.py:899
Method
test_two_concave_closed_edge
(self)
tests/test_extract_brepnet_data_from_step.py:929
Method
test_two_elliptical_edges
(self)
tests/test_extract_brepnet_data_from_step.py:955
Method
total_num_parameters
Find the total number of parameters in the network.
models/brepnet.py:574
Method
train_dataloader
(self)
models/brepnet.py:914
Method
training_step
(self, batch, batch_idx)
models/brepnet.py:678
← previous
next →
301–400 of 409, ranked by callers