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

The HybridCAD dataset is a novel resource tailored for hybrid additive-subtractive feature recognition in Computer-Aided Design (CAD) models, uniquely combining features from both manufacturing processes. Building on the MFCAD and MFCAD++ datasets, HybridCAD introduces additive manufacturing features alongside traditional subtractive ones, enabling the exploration and development of machine learning models for more complex, hybrid manufacturing applications. This dataset is especially suited for automatic feature recognition (AFR) research and model training in hybrid manufacturing contexts.

Dataset Composition

The dataset consists of 8,938 Boundary Representation (B-Rep) CAD models distributed into three main directories:

1. STEP Files:

   • Contains CAD models in STEP format, each representing a distinct part with hybrid manufacturing features.
   • Generated using PythonOCC CAD software, these CAD files serve as the foundation for feature recognition tasks.

2. Feature Labels:

   • File: feature_labels.txt
   • Provides unique label IDs for each B-Rep face in every CAD model, denoting the manufacturing feature name it belongs to.
   • Each CAD model face is labelled according to one of 29 hybrid additive-subtractive features, allowing for accurate and detailed feature recognition.
   •  

3. Hierarchical B-Rep Graphs:

   • Stored in HDF5 format for structured access to B-Rep data.
   • Detailed structural information is available in h5_structure.txt, explaining the hierarchical arrangement of B-Rep graphs.

Dataset Splits

The dataset is split into training, validation, and testing sets as follows:

• Training Set: 6,256 samples (70%)
• Validation Set: 1,342 samples (15%)
• Testing Set: 1,340 samples (15%)

Hybrid Manufacturing Features

HybridCAD includes a diverse range of additive and subtractive manufacturing features, expanding beyond the subtractive-only features of MFCAD++. This inclusion enables the exploration of hybrid manufacturing processes and the recognition of a broader feature set in CAD models. The complete feature list includes:

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      Stock