search

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.
178 stars 41 forks source link

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

Open mlbo opened 4 years ago

mlbo commented 4 years ago

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

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

image

tjuskyzhang commented 4 years ago

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.