- 博客(3)
- 收藏
- 关注
RSS订阅原创 pytorch多卡训练和单卡训练加载预训练模型时可能出现的问题
多卡训练时权重名字会带有module，有时候会加载不进去，尤其是strict=false的情况下。注释中适用多卡加载模型的情况。
2023-03-30 13:26:36
484
原创 albumentations库
今天在使用这个图像增强库albumentations的时候遇到了一个问题，由于我做的分割任务需要有两个不同的mask，我就对原始的代码进行了如下的修改： pair = self.transform(image=image, mask=mask) image, mask = pair["image"], pair["mask"] 原始的代码是这样 pair = self.transform(image=image, mask=mask, mask_a=mask_a) image, mask = pa
2022-03-29 12:38:40
664
原创 cv2.drawContours+TypeError: an integer is required (got type NoneType)
今天在复现cvpr2021的一篇文章SPOL(shallow feature matter巴拉巴拉的，具体讲的是浅层特征融合在弱监督定位中起到的作用)，遇到了这个问题，在使用cv2.drawContours时出现了Nonetype的错误。 因为cv2很容易报这种错，我一开始以为是对象读取的问题，于是逐个打印了很多变量，后来发现只要重装opencv-python就好了。我原来的版本是4.5.3，改成4.5.5之后就ok了，记录一下。 ...
2022-03-11 16:15:34
776
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人