Views
No views yet
| 文件 | 描述 | 大小 |
|---|---|---|
encoder_best.pth | 编码器模型 | ~9MB |
decoder_best.pth | 解码器模型 | ~8MB |
seg_92.pth | 分割模型 (几何校正) | ~15MB |
1from huggingface_hub import hf_hub_download
2import torch
3
4# 下载模型
5encoder_path = hf_hub_download(repo_id="ghgjcqwe/RANDmark-demo", filename="encoder_best.pth")
6decoder_path = hf_hub_download(repo_id="ghgjcqwe/RANDmark-demo", filename="decoder_best.pth")
7
8# 加载模型
9encoder = EncoderDecoder(...)
10encoder.encoder.load_state_dict(torch.load(encoder_path))
11encoder.decoder.load_state_dict(torch.load(decoder_path))