search

sophgo / tpu-mlir

Machine learning compiler based on MLIR for Sophgo TPU.
Other
613 stars 154 forks source link

算法框架 #188

Closed xc597 closed 1 week ago

xc597 commented 2 months ago

请问 yolov5 和 resnet 使用的算法框架是公司自己研发的一套吗，不是主流的 pytorch 或者 tensorflow 吗

HarmonyHu commented 1 week ago

您好，是对开源的算法进行模型转换，一般建议先导出成onnx @xc597