- 博客(4)
- 收藏
- 关注
RSS订阅原创 使用torch.jit.trace另存模型为TorchScript格式出现的问题
②检查模型导入时的设置，有些模型训练和测试调用的forward函数不同，导入模型时只需将相关参数设置成训练即可，以ssd.pytorch为例。像上述ssd返回的就是一个list，可以改成return output[0]，返回第一个值即可解决。，可以将后处理过程单独定义一个函数，模型推理完再调用该函数进行后处理；①将后处理过程（Detect、NMS等，具体看报错的是哪个函数）从。解决方式：检查模型forward函数的返回值是否为Tensor。字面解释：无法导出“Detect”函数，需要将其移除。
2024-06-04 17:11:45
643
1
原创 libgomp-d22c30c5.so、libgomp.so在aarch64出现报错的解决方式
export LD_PRELOAD=$LD_PRELOAD:libgomp-d22c30c5的具体位置。查找libgomp-d22c30c5所在位置。（降低scikit-image的版本）
2024-04-30 15:25:52
779
原创 yolov3-tiny darknet转caffe模型出现维度不匹配的问题
可以看到这两层的维度分别为（1,128,62,126）和（1,256,64,128），因此出现了top_shape[j] == bottom[i]->shape(j) (62 vs. 64)报错，这个问题有可能是maxpool参数给错导致的。4.在yolov3-tiny.prototxt找到layer12-maxpool，在yolov3-tiny.cfg找到该maxpool对应的位置，可以看到当前maxpool卷积核的大小为2，将其改成1输出就正常了。如果您也出现了类似的，可以按照上述思路试试。
2023-12-05 10:30:04
88
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人