Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- cad
|
| 5 |
+
- brep
|
| 6 |
+
- machining-feature-recognition
|
| 7 |
+
- graph-neural-network
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# BrepMFR Model Weights
|
| 11 |
+
|
| 12 |
+
Model checkpoints for [BrepMFR](https://github.com/cad-results/BrepMFR) — machining feature recognition in B-rep models through deep learning and domain adaptation.
|
| 13 |
+
|
| 14 |
+
## Files
|
| 15 |
+
|
| 16 |
+
| Path | Description | F1 Score |
|
| 17 |
+
|------|-------------|----------|
|
| 18 |
+
| `trial1_ss/best-epoch41-f1_0.9476.ckpt` | Best overall (full dataset) | 0.9476 |
|
| 19 |
+
| `trial1_ss/last.ckpt` | Last epoch, full dataset | — |
|
| 20 |
+
| `trial1_ss1500/best-epoch46-f1_0.9455.ckpt` | Best, 1500 sample subset | 0.9455 |
|
| 21 |
+
| `trial1_ss1500/last.ckpt` | Last epoch, ss1500 | — |
|
| 22 |
+
| `trial1_ss1000/best-epoch28-f1_0.9146.ckpt` | Best, 1000 sample subset | 0.9146 |
|
| 23 |
+
| `trial1_finetune_ss/best-epoch26-f1_0.9356.ckpt` | Best finetuned | 0.9356 |
|
| 24 |
+
| `pipe_classifier/best-epoch64-f1_0.5517.ckpt` | Pipe classifier (different task) | 0.5517 |
|
| 25 |
+
| `face_seg_heavy/best-epoch43-f1_0.8626.ckpt` | Face segmentation (heavy arch) | 0.8626 |
|
| 26 |
+
| `brepsimplifier/f1_0.9415.ckpt` | BrepFormer for defeaturing | 0.9415 |
|
| 27 |
+
|
| 28 |
+
## Source Code
|
| 29 |
+
|
| 30 |
+
[https://github.com/cad-results/BrepMFR](https://github.com/cad-results/BrepMFR)
|