|
发表于 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,使用图片训练好,此时的模型已经模型参数要放在哪?
|
扫一扫浏览分享
|
|
|
|
|
|
|
发表于 6 天前
你好
1.这个要看你的工作流。如果你的工作流需要用到spirecv里的功能,那就需要放到spirecv下。
2.在SpireCV/algorithm/common_det下主要是用于加载模型的,如需改动,只需针对模型的输入、输出更改代码中对应的size即可。
|
|
|
|
|
|
|
|