HybridCAD++: Expanded Dataset for Hybrid Additive-Subtractive Manufacturing Feature Recognition in B-Rep CAD Models

The HybridCAD++ dataset is a significantly expanded version of the HybridCAD dataset, offering a larger volume of CAD models and a more comprehensive set of hybrid additive-subtractive manufacturing features. This dataset includes additional feature classes, bringing the total to 36, and contains over 161,000 samples—making it a unique and robust resource for machine learning applications in hybrid manufacturing feature recognition.

Key Differences from HybridCAD

• Increased Dataset Volume: HybridCAD++ features a total of 161,441 CAD models, significantly larger than the original HybridCAD dataset.
• Expanded Feature Classes: This dataset includes 36 feature labels, with newly added classes. This increase in feature variety enhances the dataset's applicability to complex hybrid manufacturing scenarios.

Dataset Composition

The dataset includes three primary components:

1. STEP Files:

   • Each CAD model is stored in STEP format and includes labeled B-Rep faces for hybrid manufacturing feature recognition.
   • The CAD models were generated programmatically using PythonOCC, ensuring consistent quality and scalability.

2. Feature Labels:

   • File: feature_labels.txt
   • Contains label IDs for each hybrid additive-subtractive feature across B-Rep faces in each CAD model.
   • With 36 unique feature classes, this file allows precise mapping of CAD model faces to specific hybrid features.
   •  

3. Hierarchical B-Rep Graphs:

   • Stored in HDF5 format, these graphs provide structured access to B-Rep data, as detailed in h5_structure.txt.

Dataset Splits

The dataset is divided into three subsets, structured for effective model training and evaluation:

• Training Set: 113,008 samples (70%)
• Validation Set: 32,288 samples (20%)
• Testing Set: 16,145 samples (10%)

Full Feature Label List

This comprehensive list includes both subtractive and additive manufacturing features, with added classes for more intricate hybrid manufacturing applications:

Label    Feature
0    Chamfer
1    Through hole
2    Triangular passage
3    Rectangular passage
4    6-sided passage
5    Triangular through slot
6    Rectangular through slot
7    Circular through slot
8    Rectangular through step
9    2-sided through step
10    Slanted through step
11    O-ring
12    Blind hole
13    Triangular pocket
14    Rectangular pocket
15    6-sided pocket
16    Circular end pocket
17    Rectangular blind slot
18    Vertical circular end blind slot
19    Horizontal circular end blind slot
20    Triangular blind step
21    Circular blind step
22    Rectangular blind step
23    Round
24    Extrude cylinder
25    Extrude rectangle
26    Extrude triangle
27     Extrude hexagon
28     Extrude pentagon
29      Elliptical/Oval blind hole
30      Elliptical/Oval through hole
31      Slot hole
32      Obround boss
33      5-sided passage
34      5-sided pocket
35      Cylinder with hole
36      Stock