yolo-tensorrt icon indicating copy to clipboard operation
yolo-tensorrt copied to clipboard

尺寸不是1:1

Open zhangxiaopang88 opened this issue 4 years ago • 9 comments

您好,您试过尺寸不是1:1的模型吗,我试尺寸是1:1的测试成功,不是1:1的测试结果为空

zhangxiaopang88 avatar Mar 17 '20 05:03 zhangxiaopang88

@zhangxiaopang88 没有测试过,如果模型输入尺寸不是1:1,那么模型里对应的kernel步长之类的可能也要跟着改,但是非标准的可能trt不支持,你可以多测试下

enazoe avatar Mar 17 '20 05:03 enazoe

我把关于长宽不相等的断言给注掉了, // assert(inputDims.d[0] == inputDims.d[1]); assert(kernelSize.d[0] == kernelSize.d[1]); assert(stride.d[0] == stride.d[1]); assert(padding.d[0] == padding.d[1]); 其他的条件都满足,您看还需要修改哪些地方,能给个建议吗 @enazoe

zhangxiaopang88 avatar Mar 17 '20 06:03 zhangxiaopang88

个人感觉是trt不支持这种不规则的filters

enazoe avatar Mar 17 '20 07:03 enazoe

我看到输入尺寸不一致的例子,应该是支持的输入尺寸不一致的情况,我再查查资料 @enazoe

zhangxiaopang88 avatar Mar 17 '20 07:03 zhangxiaopang88

@zhangxiaopang88 I am also dealing with input size inequality, have you find any solution? (My Linux Notebook has no pinyin input method...)

xuhao1 avatar Mar 23 '20 23:03 xuhao1

If simply comment all assert, the network gives a lot of wrong detection results...

xuhao1 avatar Mar 23 '20 23:03 xuhao1

我尝试再次对比,我改动的并不大,只是涉及GridSize的地方改成宽高,处理1:1输入的话不报异常,但是不是1:1的话就会有问题,我准备把yolo转成onnx试试能不能成功

zhangxiaopang88 avatar Mar 26 '20 02:03 zhangxiaopang88

Hey @zhangxiaopang88 and @xuhao1, I also have inequal sized input images. Have you any progress on that issue?

aybora avatar May 12 '20 18:05 aybora

reference this commit 2edde41546bd58f13f2f51c4c9ecbe966f3267a8 @aybora @xuhao1 @zhangxiaopang88

enazoe avatar Aug 27 '20 14:08 enazoe