ly17 commited on
Commit
352473c
·
verified ·
1 Parent(s): e260cb2

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +43 -0
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # TC-SemiSAM3-new
2
+
3
+ Semi-supervised SAM3 model for coronary vessel segmentation.
4
+
5
+ ## Model Description
6
+
7
+ This is the final checkpoint from semi-supervised training using:
8
+ - 5 labeled videos
9
+ - 31 unlabeled videos
10
+ - Mean Teacher framework with temporal consistency
11
+
12
+ ## Usage
13
+
14
+ ```python
15
+ import torch
16
+
17
+ # Load checkpoint
18
+ checkpoint = torch.load("checkpoint_final.pt", map_location="cpu")
19
+
20
+ # Get student model weights
21
+ state_dict = checkpoint["student_state_dict"]
22
+
23
+ # Load into SAM3 model
24
+ model.load_state_dict(state_dict, strict=False)
25
+ ```
26
+
27
+ ## Training Details
28
+
29
+ - Framework: Mean Teacher + Temporal Consistency
30
+ - Labeled data: 5 videos (140 frames)
31
+ - Unlabeled data: 31 videos
32
+ - Confidence-aware regularization enabled
33
+
34
+ ## Performance
35
+
36
+ | Dataset | Dice | clDice |
37
+ |---------|------|--------|
38
+ | 36 Training Videos | 0.6811 | 0.7492 |
39
+ | 36 Videos (Adaptive) | 0.7232 | 0.7775 |
40
+
41
+ ## Related
42
+
43
+ - GitHub: https://github.com/qimingfan10/TC-SemiSAM.git