常见ML任务的模型和模型实用程序

xt-models的Python项目详细描述


xt型号

说明

这个repo包含处理ML任务的通用模型和实用程序,由Xtract AI开发。在

还有更多。在

安装

来自PyPi:

pip install xt-models

来源:

^{pr2}$

使用

获取分割模型

fromxt_models.modelsimportModelBuilder,SegmentationModulefromtorchimportnndeep_sup_scale=0.4fc_dim=2048n_class=2net_encoder=ModelBuilder.build_encoder(arch="resnet50dilated",fc_dim=fc_dim,weights="/nasty/scratch/common/smart_objects/model/ade20k/encoder_epoch_20.pth")net_decoder=ModelBuilder.build_decoder(arch="ppm_deepsup",fc_dim=fc_dim,num_class=150,weights="/nasty/scratch/common/smart_objects/model/ade20k/decoder_epoch_20.pth")in_channels=net_decoder.conv_last[-1].in_channelsnet_decoder.conv_last[-1]=nn.Conv2d(in_channels,n_class,kernel_size=(1,1),stride=(1,1))net_decoder.conv_last_deepsup=nn.Conv2d(in_channels,n_class,1,1,0)model=SegmentationModule(net_encoder,net_decoder,deep_sup_scale)

获取检测模型

from xt_models.models import Model
import torch

# Load a fine-tuned model for inference
model_name = "yolov5x"
model = Model(model_name,nc=15)
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
weights = "/nasty/scratch/common/smart_objects/model/veh_detection/yolov5_ft/best_state_dict.pt"
ckpt = torch.load(weights, map_location=device)
model.load_state_dict(ckpt['model_state_dict'])

# Load pre-trained COCO model for finetuning/inference
model_name = "yolov5x"
model = Model(model_name,nc=80)
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
weights = "/nasty/scratch/common/smart_objects/model/veh_detection/yolov5_pretrain/yolov5x_state_dict.pt"
ckpt = torch.load(weights, map_location=device)
model.load_state_dict(ckpt['model_state_dict'])
# Fine-tuning number of classes
n_class = 15
model.nc = n_class

实施新模型

如果您总是要为不同的项目复制和粘贴相同的模型代码,只需将模型代码添加到models目录中,并将其导入models/__init__.py文件。在

数据源

[数据说明和链接]

依赖关系/许可

[依赖项及其许可证的列表,包括数据]

参考文献

[参考文献列表]

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java如何使用bouncycastle从pkcs7文件获取证书?   java是向字符串数字添加零的有效方法吗?   Android在外部播放器中播放视频url   如何在Java代码中调用Wildfly事务管理器?   java为什么在使用键盘完成选择时不触发选择事件?   java xpath错误:jaxb外部绑定文件上为null   java如何避免在活动之间切换时出现白屏   Hadoop安装程序中的java“unknown.prolexic.com:未知错误”   要用作out参数的java自定义数字类   java如何向spring控制器的http请求添加属性?   for循环中的java静态值   java为图像编辑器快速、可靠地保存/加载文档状态   java数组越界和空指针异常   java如何在列表<Object>对象包含对象的对象上检查null?   java从URL下载文件到tomcat服务器   GlassFish“ogs3.1.2.2windows.exe”和“ogs3.1.2.2windowsml.exe”文件之间的java差异   无MathLab的Java Gabor图像处理   包含CSV中不存在的其他字段的java mongoimport