备份吧 关注:351贴子:3,178
  • 0回复贴,共1

导入模型代码备份

只看楼主收藏回复

net = modeling._segm_resnet('deeplabv3plus', 'resnet101', num_classes=2, output_stride=8, pretrained_backbone="C:/best_deeplabv3plus_resnet101_voc_os16.pth").cuda()
#net.load_from(weights=np.load("./best_deeplabv3plus_resnet101_voc_os16.pth"))


IP属地:湖北1楼2024-04-15 15:27回复