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

<p>The <em>HybridCAD</em> 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 <em>MFCAD</em> and <em>MFCAD++</em> datasets, <em>HybridCAD</em> 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.</p> <h3>Dataset Composition</h3> <p>The dataset consists of 8,938 Boundary Representation (B-Rep) CAD models distributed into three main directories:</p> <ol> <li> <p><strong>STEP Files</strong>:</p> <ul> <li>Contains CAD models in STEP format, each representing a distinct part with hybrid manufacturing features.</li> <li>Generated using PythonOCC CAD software, these CAD files serve as the foundation for feature recognition tasks.</li> </ul> </li> <li> <p><strong>Feature Labels</strong>:</p> <ul> <li><strong>File</strong>: <code>feature_labels.txt</code></li> <li>Provides unique label IDs for each B-Rep face in every CAD model, denoting the manufacturing feature name it belongs to.</li> <li>Each CAD model face is labelled according to one of 29 hybrid additive-subtractive features, allowing for accurate and detailed feature recognition.</li> </ul> <ul> <li>&nbsp;</li> </ul> </li> <li> <p><strong>Hierarchical B-Rep Graphs</strong>:</p> <ul> <li>Stored in HDF5 format for structured access to B-Rep data.</li> <li>Detailed structural information is available in <code>h5_structure.txt</code>, explaining the hierarchical arrangement of B-Rep graphs.</li> </ul> </li> </ol> <h3>Dataset Splits</h3> <p>The dataset is split into training, validation, and testing sets as follows:</p> <ul> <li><strong>Training Set</strong>: 6,256 samples (70%)</li> <li><strong>Validation Set</strong>: 1,342 samples (15%)</li> <li><strong>Testing Set</strong>: 1,340 samples (15%)</li> </ul> <h3>Hybrid Manufacturing Features</h3> <p><em>HybridCAD</em> includes a diverse range of additive and subtractive manufacturing features, expanding beyond the subtractive-only features of <em>MFCAD++</em>. 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:</p> <p>Label &nbsp; &nbsp;Feature<br>0 &nbsp; &nbsp;Chamfer<br>1 &nbsp; &nbsp;Through hole<br>2 &nbsp; &nbsp;Triangular passage<br>3 &nbsp; &nbsp;Rectangular passage<br>4 &nbsp; &nbsp;6-sided passage<br>5 &nbsp; &nbsp;Triangular through slot<br>6 &nbsp; &nbsp;Rectangular through slot<br>7 &nbsp; &nbsp;Circular through slot<br>8 &nbsp; &nbsp;Rectangular through step<br>9 &nbsp; &nbsp;2-sided through step<br>10 &nbsp; &nbsp;Slanted through step<br>11 &nbsp; &nbsp;O-ring<br>12 &nbsp; &nbsp;Blind hole<br>13 &nbsp; &nbsp;Triangular pocket<br>14 &nbsp; &nbsp;Rectangular pocket<br>15 &nbsp; &nbsp;6-sided pocket<br>16 &nbsp; &nbsp;Circular end pocket<br>17 &nbsp; &nbsp;Rectangular blind slot<br>18 &nbsp; &nbsp;Vertical circular end blind slot<br>19 &nbsp; &nbsp;Horizontal circular end blind slot<br>20 &nbsp; &nbsp;Triangular blind step<br>21 &nbsp; &nbsp;Circular blind step<br>22 &nbsp; &nbsp;Rectangular blind step<br>23 &nbsp; &nbsp;Round<br>24 &nbsp; &nbsp;Extrude cylinder<br>25 &nbsp; &nbsp;Extrude rectangle<br>26 &nbsp; &nbsp;Extrude triangle<br>27 &nbsp; &nbsp; Extrude hexagon<br>28 &nbsp; &nbsp; Extrude pentagon<br>29 &nbsp; &nbsp; &nbsp;Stock</p>