warning fix
This commit is contained in:
+2
-2
@@ -22,10 +22,10 @@ int main(int argc, char *argv[]) {
|
|||||||
signal(SIGINT, sig_handler);
|
signal(SIGINT, sig_handler);
|
||||||
|
|
||||||
|
|
||||||
char *net = "yolo3_berkeley.rt";
|
std::string net = "yolo3_berkeley.rt";
|
||||||
if(argc > 1)
|
if(argc > 1)
|
||||||
net = argv[1];
|
net = argv[1];
|
||||||
char *input = "../demo/yolo_test.mp4";
|
std::string input = "../demo/yolo_test.mp4";
|
||||||
if(argc > 2)
|
if(argc > 2)
|
||||||
input = argv[2];
|
input = argv[2];
|
||||||
char ntype = 'y';
|
char ntype = 'y';
|
||||||
|
|||||||
@@ -48,7 +48,7 @@ public:
|
|||||||
|
|
||||||
bool addLayer(Layer *l);
|
bool addLayer(Layer *l);
|
||||||
void print();
|
void print();
|
||||||
const char *getNetworkRTName(char *network_name);
|
const char *getNetworkRTName(const char *network_name);
|
||||||
|
|
||||||
cudnnDataType_t dataType;
|
cudnnDataType_t dataType;
|
||||||
cudnnTensorFormat_t tensorFormat;
|
cudnnTensorFormat_t tensorFormat;
|
||||||
|
|||||||
@@ -133,7 +133,7 @@ void CenternetDetection::preprocess(cv::Mat &frame){
|
|||||||
float new_height = sz.height * scale;
|
float new_height = sz.height * scale;
|
||||||
float new_width = sz.width * scale;
|
float new_width = sz.width * scale;
|
||||||
if(sz.height != sz_old.height && sz.width != sz_old.width){
|
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];
|
float s[2];
|
||||||
|
|
||||||
if(sz.width > sz.height){
|
if(sz.width > sz.height){
|
||||||
|
|||||||
+1
-1
@@ -120,7 +120,7 @@ void Network::print() {
|
|||||||
printCenteredTitle("", '=', 60);
|
printCenteredTitle("", '=', 60);
|
||||||
std::cout<<"\n";
|
std::cout<<"\n";
|
||||||
}
|
}
|
||||||
const char *Network::getNetworkRTName(char *network_name){
|
const char *Network::getNetworkRTName(const char *network_name){
|
||||||
networkName = network_name;
|
networkName = network_name;
|
||||||
int network_name_len = strlen(network_name);
|
int network_name_len = strlen(network_name);
|
||||||
char *RTName = (char *)malloc((network_name_len + 9)*sizeof(char));
|
char *RTName = (char *)malloc((network_name_len + 9)*sizeof(char));
|
||||||
|
|||||||
Reference in New Issue
Block a user