对于原始Yolov5网络的后处理部分的逻辑,Cambricon-PyTorch直接使用一个大的BANGC算子完成 ... detect_out = torch.ops.torch_mlu.yolov5_detection_output(output[0], ...
確定! 回上一頁