![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
nnunet
4月16!
平时记录用,不喜忽喷
展开
-
nnunetv2 .pth 转 .onnx (Tag=2.0/2.1)
转化为onnx后可以转化为engine文件,方便在c++使用;(可以移步tensorRT分类中看)nnunet v2版本的模型转化为onnx。原创 2023-11-30 11:03:31 · 794 阅读 · 16 评论 -
surface dice
nnUNet中的设计surface dice的计算方法:import numpy as npfrom medpy.metric.binary import __surface_distancesdef normalized_surface_dice(a: np.ndarray, b: np.ndarray, threshold: float, spacing: tuple = None, connectivity=1): """ This implementation dif.原创 2022-04-12 14:08:43 · 1357 阅读 · 0 评论 -
nnunetv1 .model to .onnx
动态输入batch_size:就是在末尾加上 dynamic_axes = {'input': {0: 'batch_size'},'output': {0: 'batch_size'}}为什么是inputoutput因为我的输入和输出都只有一个,且名字是我个人命名为 input_names=['input'], output_names=['output'],import torchfrom torch import nnimp...原创 2022-03-10 14:57:40 · 1136 阅读 · 2 评论