Micaela Verucchi
907df27e07
- Fix monodepth2 network, now works with both cuDNN and tensorRT - Substitute cuDNN ELU with tkDNN one - add DepthNN class - add demoDepth demo, now only works with monodepth2 net Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com> Francesco Gatti <gattifrancesco@hotmail.it>
13 lines
416 B
C++
13 lines
416 B
C++
#pragma once
|
|
#include <iostream>
|
|
#include <opencv2/core/types.hpp>
|
|
#include "tkdnn.h"
|
|
|
|
namespace tk { namespace dnn {
|
|
|
|
cv::Mat vizFloat2colorMap(cv::Mat map, double min=0, double max=0, int classes=19);
|
|
cv::Mat vizData2Mat(dnnType *dataInput, tk::dnn::dataDim_t dim, int img_h, int img_w, double min=0, double max=0, int classes=0);
|
|
cv::Mat vizLayer2Mat(tk::dnn::Network *net, int layer, int imgdim = 1000);
|
|
|
|
}}
|