目标检测ssd如何修改输入图像的比例为非正方形
收藏
刚使用paddplepaddle框架,做目标识别用,使用自带的ssd的训练网络ssd_vgg16_300.yml/ssd_vgg16_512.yml,因为我们使用场景针对的都是16:9的图片,因此想把网络输入的输入的尺寸改为非正方形的尺寸比例,哪位大哥可以分享一下如何修改?之前用caffe每层网络参数都是直接可见,将输入及中间各种相关的修改了就ok了,paddlepaddle没搞明天那些地方需要修改,只改ssd_vgg16_300.yml试了下不够。
0
收藏
请登录后评论
在ssd_vgg16_300.yml里面只修改 image_shape: [3, 300, 300] 测试是不行的,有哪位知道怎么修改? 只需要修改ssd_vgg16_300.yml还是说还有其它对应的地方需要修改。
应该使用的是PaddleDetection吧? 如要要自定义数据集,可以参考 数据处理模块文档:https://github.com/PaddlePaddle/PaddleDetection/blob/release/0.3/docs/advanced_tutorials/READER.md。建议使用 方式一:将数据集转换为COCO格式。这样处理起来更方便些。
如果是单纯想控制输入的图像直接resize就好了
我们自己定义了训练测试数据集,但是我们的图片都是16:9的尺寸,最终应用的时候输入的也是16:9的尺寸。所以希望修改训练网络让输入的格式是16:9的,而不是正方形的,训练出来的模型不是非常理想,想着是不是可以不用正方形的图片训练。
简单的方法是将现有的图片拼一块白的变成正方形的图片,应用的试试也是这样操作,这就多出不少没用的计算,最好的方法是能针对性的修改网络。