paddeldetection多波段数据目标检测
收藏
刚开始用paddeldetection,RGB数据开箱即用,感觉良好
最近做论文,遇到一个需求,数据有十几个波段,高光谱,遇到问题
1、模型入口input数据的输入波段在哪设置,RGB是3,需要改成16,在哪改?
2、重写数据读取模块,类似pytorch的dataset模块,paddeldetection的在哪个文件?
从pytorch转过来,刚开始用,问题比较基础,希望得到帮助。
0
收藏
请登录后评论
刚看了下,在 train.py下基本是读取了cfg文件,然后进行配置,配置文件中没有input关键字
ppdet/modeling/backbones下, 主干网络定义的时候,有个def __call__(self, input):
这个input应该就是网络的入口了,更改即可
奋战了几个小时,找到了另一个关键点
paddledetection读取数据并非在dataset中进行,而是在数据增强过程中进行
so,更改数据增强代码即可实现数据读取模块的重写
吐槽下,这论坛完全没人气啊啊啊啊啊啊
很有帮助 感谢