AllentDan

点击[这里](https://allentdan.github.io/2020/12/16/pytorch%E9%83%A8%E7%BD%B2torchscript%E7%AF%87/#more)的一些报错部分

Please ensure your CUDA toolkit is installed and the environment variable is set.

Hi, 回得慢不好啥意思哈。计时如果有用cuda的话，需要先synchronize再计算会准确。此外模型一般前期的几次推理会慢，几次后会速度趋于稳定

> 谢谢回复哈！我这出现的情况更不同，第一次推理很快，而后约20次推理非常慢，之后推理速度才正常快了 没遇到过😂

Slightly different from IoU actually as IoU is not differentiable.

谢谢，最近都没时间做。后面空下来了可以试试。额，看源码？

Define a onnx model with the exact same structure as libtorch model and copy the weight like convolution to the onnx model.

head是公用的，不只是Unet要用。另外如果upsample输入输出分辨率相同，计算与否不影响。 dice 计算部分仔细看看，inter部分计算的是概率图乘以mask再求和。

> 另外一个问题： > > * labelme的标注数据转mask时，为什么不直接绘制成单通道mask(比如填充的时候就不用name2color填充，而用name2index填充)，而是要先绘制彩色mask然后再转单通道mask？ 单通道的Resize后可能出现物体边缘变成其他类，彩色的可能小很多。

May fix the lint error and resolve conflicts.