V8heart/CAPSTONE-gnn-weights

CAPSTONE GNN Weights

Checkpoints for CAPSTONE powerline detection (YOLinO + GNN two-stage pipeline).

Best model (512×512)

Stage 2 final: exp83_gnn_ttpla_512512_from_exp80/ep0006_model.pth (epoch 6)

Warm-starts from Stage 1 exp80_ttpla_512512_scale16/best_model.pth. Config: configs/experiments/exp83_gnn_ttpla_512512_from_exp80.yaml.

Reference (previous Stage 2): exp81_gnn_ttpla_512512_from_exp80/ep0058_model.pth

All checkpoints

Path	Experiment	Role
exp80_ttpla_512512_scale16/best_model.pth	exp80	Stage 1 geometry (512²)
exp83_gnn_ttpla_512512_from_exp80/ep0006_model.pth	exp83 ep6	Stage 2 GNN (512², final)
exp81_gnn_ttpla_512512_from_exp80/ep0058_model.pth	exp81 ep58	Previous Stage 2 (reference)
exp19_fpn_bottomup_p4_num_predictors4_1024/ep0042_model.pth	exp19 ep42	1024² ablation geometry
exp76_gnn_ttpla_full/ep0005_model.pth	exp76 ep5	1024² ablation GNN
exp23_finetune_kepco/best_model.pth	exp23	KEPCO geometry fine-tune
exp71_gnn_kepco/best_model.pth	exp71	KEPCO GNN

Datasets

Dataset	Hugging Face
TTPLA benchmark (512²)	V8heart/yolino-ttpla-benchmark
TTPLA ablation tiles (1024²)	V8heart/ttpla-yolino-1024
KEPCO cross-domain	V8heart/Yolino-KEPCO

Download

# Final Stage 2 model (exp83, epoch 6)
hf download V8heart/CAPSTONE-gnn-weights \
  exp83_gnn_ttpla_512512_from_exp80/ep0006_model.pth \
  --repo-type model \
  --local-dir ttpla_train_exp/log/checkpoints/exp83_gnn_ttpla_512512_from_exp80

# Stage 1 geometry
hf download V8heart/CAPSTONE-gnn-weights \
  exp80_ttpla_512512_scale16/best_model.pth \
  --repo-type model \
  --local-dir ttpla_train_exp/log/checkpoints/exp80_ttpla_512512_scale16

Usage

--explicit_model log/checkpoints/exp83_gnn_ttpla_512512_from_exp80/ep0006_model.pth

See the CAPSTONE README for training, inference, and evaluation commands.