6

阿木币

0

精华

6 小时

在线时间

应届白菜

Rank: 1

发表于 2024-11-12 16:25:37 231 浏览 1 回复

更改模型网络结构,放在spirecv哪里?

1,spirecv所用的都是跨平台封装,如果我不想使用跨平台封装,只是利用pytorch实现一个CNN,进行特征提取,而不是目标检测,那么我训练好的CNN模型,以及模型参数是要放在spirecv下吗?还是和没有spirecv的时候那样,直接放在prometheus下?

诉求:能看到模型的完整代码,因为spirecv是封装好的,不容易学习,且无法改变模型。比如说改变的yolov5,在SpireCV/algorithm/common_det下不方便改变模型,我更想download下来yolov5,使用图片训练好,此时的模型已经模型参数要放在哪?



扫一扫浏览分享
回复

使用道具 举报

4

阿木币

0

精华

10 小时

在线时间

技术大V

Rank: 4

发表于 6 天前
你好

1.这个要看你的工作流。如果你的工作流需要用到spirecv里的功能,那就需要放到spirecv下。

2.在SpireCV/algorithm/common_det下主要是用于加载模型的,如需改动,只需针对模型的输入、输出更改代码中对应的size即可。
回复 点赞

使用道具 举报

返回列表
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表