Commit Graph

4 Commits

Author SHA1 Message Date
书包
310c139cae Update upsample_layer.cpp
replace two matrix multiplication to resize layer, to adapt non square input size.
2021-05-21 08:51:28 +08:00
Pieris Kalligeros
8d5847c658 Comment leftover assertions of square dims 2021-02-19 16:51:37 +02:00
Pieris Kalligeros
59b9b9498a Update yoloPlugins.h
Corrected gridSize mistake in signature of YoloLayer class causing error at compile time.
2021-02-19 16:03:38 +02:00
Marcos Luciano
9dbb9edb7f Added non_square
Support for non square YOLO models.
2021-02-18 18:01:03 -03:00