segmentation_pytorch库需要用huggingface,但是国内无法访问时,使用离线的方式

方法一:

使用这种方式FPN(encoder=maxvit_small_tf_224(pretrained_cfg_overlay=dict(file=cached_safe_file),features_only=True,pretrained=False),读取bin文件

python 复制代码
cached_safe_file = "/home/xx/myProject/data/pytorch_weights/maxvit_small_tf_224.bin"
# args.net.encoder.model = timm.create_model(
#     "maxvit_base_tf_224",pretrained=True,
#     pretrained_cfg_overlay=dict(file=cached_safe_file),features_only=True)

from Netwroks.segmentation.segmentation.decoders.fpn.transformer_decoder import FPN
from Netwroks.segmentation.encoder.model2 import maxvit_base_tf_224,maxvit_tiny_tf_224,maxvit_small_tf_224

args.net = FPN(encoder=maxvit_small_tf_224(pretrained_cfg_overlay=dict(file=cached_safe_file),features_only=True,pretrained=False),
    in_channels=3,      
    classes=1, 
    activation='sigmoid' )

方法二

下面这种方式也可以,读取model.safetensors

python 复制代码
args.net  = smp.FPN(
    encoder_name="tu-maxvit_base_tf_224",#'tu-maxvit_base_tf_224',        # 选择解码器, 例如 mobilenet_v2 或 efficientnet-b7
    encoder_weights=None,     # 使用预先训练的权重imagenet进行解码器初始化
    in_channels=3,                  # 模型输入通道(1个用于灰度图像,3个用于RGB等)
    classes=1, 
    activation='sigmoid'                    # 模型输出通道(数据集所分的类别总数)
)
import timm
cached_safe_file = "/home/xx/.cache/huggingface/hub/models--timm--maxvit_base_tf_224.in1k/snapshots/ede2304fc169e23779075c092a477302f23660f8/model.safetensors"
args.net.encoder.model = timm.create_model(
    "maxvit_base_tf_224",pretrained=True,
    pretrained_cfg_overlay=dict(file=cached_safe_file),features_only=True)
相关推荐
烛阴6 小时前
简单入门Python装饰器
前端·python
好开心啊没烦恼7 小时前
Python 数据分析:numpy,说人话,说说数组维度。听故事学知识点怎么这么容易?
开发语言·人工智能·python·数据挖掘·数据分析·numpy
面朝大海,春不暖,花不开7 小时前
使用 Python 实现 ETL 流程:从文本文件提取到数据处理的全面指南
python·etl·原型模式
2301_805054568 小时前
Python训练营打卡Day59(2025.7.3)
开发语言·python
万千思绪8 小时前
【PyCharm 2025.1.2配置debug】
ide·python·pycharm
微风粼粼10 小时前
程序员在线接单
java·jvm·后端·python·eclipse·tomcat·dubbo
云天徽上10 小时前
【PaddleOCR】OCR表格识别数据集介绍,包含PubTabNet、好未来表格识别、WTW中文场景表格等数据,持续更新中......
python·ocr·文字识别·表格识别·paddleocr·pp-ocrv5
你怎么知道我是队长10 小时前
python-input内置函数
开发语言·python
叹一曲当时只道是寻常11 小时前
Python实现优雅的目录结构打印工具
python
hbwhmama12 小时前
python高级变量XIII
python