Tencent / ncnn

ncnn is a high-performance neural network inference framework optimized for the mobile platform
Other
20.56k stars 4.18k forks source link

image warp #2911

Open fedral opened 3 years ago

fedral commented 3 years ago

你好,请问 low-level 视频任务 常用的 deformable Conv, 基于光流 image warp 的op NCNN支持么? 谢谢!

nihui commented 3 years ago

可以使用自定义op实现,比如 https://github.com/nihui/rife-ncnn-vulkan 工程中自定义op实现光流的warp

tyrionchen commented 5 months ago

可以使用自定义op实现,比如 https://github.com/nihui/rife-ncnn-vulkan 工程中自定义op实现光流的warp

你好,请问下使用onnx2ncnn工具转换模型时碰到ncnn不支持的op应该怎么处理呢? 例如,某个rife模型的示例工程实现了基于光流 image warp(依赖GridSample) 的op,我想把这个模型转为ncnn的模型文件以便在rife-ncnn-vulkan工程中使用,但是我发现onnx2ncnn不支持GridSample。

我看有人(链接)会在导出模型时先把pytorch工程里的算子替换掉,之后再编辑.param文件把算子改回去,不知道这种方法是不是标准的操作。