Fix saving result video in demoDepth, add automatic download for monodepth2 weights

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2022-01-19 10:22:06 -08:00
parent 907df27e07
commit decd73d298
3 changed files with 16 additions and 5 deletions
+4 -2
View File
@@ -66,11 +66,13 @@ const char* output_bin[] = {
"monodepth2/debug/outputs/output-disp-3.bin"
};
const char* input_monodepth2_bin[] = {"monodepth2/debug/input.bin","monodepth2/debug/input2.bin"};
const char* input_bin = "monodepth2/debug/input.bin";
int main(){
downloadWeightsifDoNotExist(input_bin, "monodepth2", "https://cloud.hipert.unimore.it/s/iYw9QwgP6CsqxLR/download");
tk::dnn::dataDim_t dim(1,3,192,640,1);
tk::dnn::Network net(dim);
@@ -204,7 +206,7 @@ int main(){
dnnType *data;
dnnType *input_H;
readBinaryFile(input_monodepth2_bin[0],dim.tot(),&input_H,&data);
readBinaryFile(input_bin, dim.tot(),&input_H,&data);
std::cout<<"INPUT DIMENSIONS : "<<dim.tot()<<std::endl;
net.print();