输出模型每层的耗时及GFlops的代码，在yolov5和yolov7官方代码yolo.py已经实现。
在模型改进和轻量化的时候，可以观测哪一层比较耗时以及参数量比较大
代码
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--cfg', type=输出模型每层的耗时和GFlops(yolov5 yolov7)
猜你喜欢
转载自blog.csdn.net/weixin_38346042/article/details/129773690
今日推荐
周排行