search

walktree / libtorch-yolov3

A Libtorch implementation of the YOLO v3 object detection algorithm
450 stars 131 forks source link

请问这个一定得用libtorch重写yolov3么，能不能用torch.jit.trace导出pytorch的yolov3模型进行调用 #51

Closed zeta7337 closed 3 years ago

zeta7337 commented 4 years ago

请问这个一定得用libtorch重写yolov3么，能不能用torch.jit.trace导出pytorch的yolov3模型进行调用

walktree commented 3 years ago

YOLO 后半部分采用的是非标准化算子，无法直接导出。前面一部分结构是可以导出的。