diff --git a/demo/detection/detection.cpp b/demo/detection/detection.cpp index 5374846..98751af 100644 --- a/demo/detection/detection.cpp +++ b/demo/detection/detection.cpp @@ -87,7 +87,18 @@ void compute_image( cv::Mat imageORIG, int main(int argc, char *argv[]) { - if(argc < 2 || !fileExist(argv[1])) + if(argc < 3) { + std::cout<<"usage: "< validation_list.txt format: \n" + <<" path/to/image.jpg path/to/label.txt\n" + <<"> label.txt format: \n" + <<" \n" + <<" x and y are the box center, " + <<"all values are relative to the image size\n\n"; + return 1; + } + + if(!fileExist(argv[1])) FatalError("unable to read serialRT file"); //convert network to tensorRT