diff --git a/demo/demo/demo.cpp b/demo/demo/demo.cpp index ccc5c7f..75012b6 100644 --- a/demo/demo/demo.cpp +++ b/demo/demo/demo.cpp @@ -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'; diff --git a/include/tkDNN/Network.h b/include/tkDNN/Network.h index 07924d0..eb17ea8 100644 --- a/include/tkDNN/Network.h +++ b/include/tkDNN/Network.h @@ -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; diff --git a/src/CenternetDetection.cpp b/src/CenternetDetection.cpp index 997989e..97668e6 100644 --- a/src/CenternetDetection.cpp +++ b/src/CenternetDetection.cpp @@ -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){ diff --git a/src/Network.cpp b/src/Network.cpp index cbbdc91..885ec1a 100644 --- a/src/Network.cpp +++ b/src/Network.cpp @@ -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));