search

mkang315 / RCS-YOLO

[MICCAI'23] Official implementation of "RCS-YOLO: A Fast and High-Accuracy Object Detector for Brain Tumor Detection".
GNU General Public License v3.0
76 stars 10 forks source link

onnx导出失败 #14

Open ZHIZIHUABU opened 4 months ago

ZHIZIHUABU commented 4 months ago

rcs3-yolo模型无法导出onnx，报错信息为F.avg_pool2d()的kernel_size参数类型时tensor，不是int型。查询到SEBlock模块中有一个F.avg_pool2d()算子,kernel_size=input_size(3)。

mkang315 commented 4 months ago

Please look at the answer from this link Problem with F.avg_pool2d #34743. Thanks!