aghoraguru commited on
Commit
fb7fa13
·
verified ·
1 Parent(s): b73b5a3

Add tram-v2 model card

Browse files
Files changed (1) hide show
  1. README.md +58 -0
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-4.0
3
+ tags:
4
+ - object-detection
5
+ - instance-segmentation
6
+ - mask-rcnn
7
+ - swin-transformer
8
+ - sonar
9
+ - underwater
10
+ library_name: pytorch
11
+ pipeline_tag: object-detection
12
+ ---
13
+
14
+ # TRAM-V2
15
+
16
+ Reference checkpoint for **TRAM-V2** from the paper *TRAM:
17
+ Transformer-Based Mask R-CNN Framework for Underwater Object Detection in
18
+ Side-Scan Sonar Data* (Makam, Sundaram, & Sundaram).
19
+
20
+ - **Architecture**: MST (Swin-Base) + FPN + CBAM + Mask R-CNN
21
+ - **Input**: 224×224 RGB
22
+ - **Training data**: SSS_OD-5 (SeabedObjects-KLSG-derived, plane + ship)
23
+ - **Best epoch**: 27 / 47
24
+ - **Random seed**: not fixed
25
+
26
+ ## Validation metrics (this run)
27
+
28
+ | Metric | This run | Paper |
29
+ |---|---|---|
30
+ | Det mAP@0.5 | 0.8356 | 0.8321 |
31
+ | Det mAP@0.5:0.95 | 0.5496 | 0.5293 |
32
+ | Seg mAP@0.5 | 0.7538 | 0.7352 |
33
+ | Seg mAP@0.5:0.95 | 0.4244 | 0.4412 |
34
+
35
+ ## Usage
36
+
37
+ ```bash
38
+ git clone -b final-tram-v123 https://github.com/CrypticCortex/iisc-sss-codes.git
39
+ cd iisc-sss-codes
40
+ pip install -r final/requirements.txt
41
+
42
+ python -m final.tram_v2.inference \
43
+ --weights /path/to/tram_v2_best.pth \
44
+ --data-root /path/to/SSS_OD-5/valid \
45
+ --output-dir runs/tram-v2/inference \
46
+ --evaluate-map
47
+ ```
48
+
49
+ ## Files
50
+
51
+ - `tram_v2_best.pth` — best checkpoint (highest val bbox mAP@0.5:0.95)
52
+ - `training.log` — per-epoch training/validation log
53
+
54
+
55
+ ## Source
56
+
57
+ - Code: https://github.com/CrypticCortex/iisc-sss-codes/tree/final-tram-v123/final
58
+ - Other variants: [tram-v1](https://huggingface.co/aghoraguru/tram-v1) · [tram-v2](https://huggingface.co/aghoraguru/tram-v2) · [tram-v3](https://huggingface.co/aghoraguru/tram-v3)