Giters

tjuskyzhang / Scaled-YOLOv4-TensorRT

Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

my imagine size is 1024*1024.How to modify the size of an input picture。

mlbo opened this issue · comments

commented

when I modify code in yololayer.h i met such error

static constexpr int INPUT_H = 1024;
static constexpr int INPUT_W = 1024;

image

commented

when I modify code in yololayer.h i met such error

static constexpr int INPUT_H = 1024;
static constexpr int INPUT_W = 1024;

image

The "INPUT_H" and "INPUT_W" are the input size of the network, according to your's yolov4-tiny.cfg. The input image can be any size ,and will be resized into the input size of the network at the preprocessing stage. By the way, I also updated my code (the Dims of the "route_lhalf" layers in "yolov4-tiny.cpp") in order to improve the adaptability of the input size of the network.