warning fix

This commit is contained in:
Francesco Gatti
2020-04-09 01:00:14 +02:00
parent 2635d24855
commit ed0596a52c
4 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -22,10 +22,10 @@ int main(int argc, char *argv[]) {
signal(SIGINT, sig_handler);
char *net = "yolo3_berkeley.rt";
std::string net = "yolo3_berkeley.rt";
if(argc > 1)
net = argv[1];
char *input = "../demo/yolo_test.mp4";
std::string input = "../demo/yolo_test.mp4";
if(argc > 2)
input = argv[2];
char ntype = 'y';
+1 -1
View File
@@ -48,7 +48,7 @@ public:
bool addLayer(Layer *l);
void print();
const char *getNetworkRTName(char *network_name);
const char *getNetworkRTName(const char *network_name);
cudnnDataType_t dataType;
cudnnTensorFormat_t tensorFormat;
+1 -1
View File
@@ -133,7 +133,7 @@ void CenternetDetection::preprocess(cv::Mat &frame){
float new_height = sz.height * scale;
float new_width = sz.width * scale;
if(sz.height != sz_old.height && sz.width != sz_old.width){
float c[] = {new_width / 2.0, new_height /2.0};
float c[] = {new_width / 2.0f, new_height /2.0f};
float s[2];
if(sz.width > sz.height){
+1 -1
View File
@@ -120,7 +120,7 @@ void Network::print() {
printCenteredTitle("", '=', 60);
std::cout<<"\n";
}
const char *Network::getNetworkRTName(char *network_name){
const char *Network::getNetworkRTName(const char *network_name){
networkName = network_name;
int network_name_len = strlen(network_name);
char *RTName = (char *)malloc((network_name_len + 9)*sizeof(char));