warning fix
This commit is contained in:
+2
-2
@@ -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';
|
||||
|
||||
@@ -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;
|
||||
|
||||
@@ -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
@@ -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));
|
||||
|
||||
Reference in New Issue
Block a user