CenterNet2TorchScript
所属分类：数值算法/人工智能
开发工具：Cuda
文件大小：36KB
下载次数：0
上传日期：2020-02-26 09:58:26
上 传 者：
sh-1993
说明： CenterNet2TorchScript，中心网pytorch模型到torch脚本模型
(CenterNet2TorchScript,centernet pytorch model to torch script model)
文件列表:
DCNv2 (0, 2020-02-26)
DCNv2\LICENSE (1520, 2020-02-26)
DCNv2\__init__.py (0, 2020-02-26)
DCNv2\dcn_v2.py (10679, 2020-02-26)
DCNv2\make.sh (51, 2020-02-26)
DCNv2\setup.py (1977, 2020-02-26)
DCNv2\src (0, 2020-02-26)
DCNv2\src\cpu (0, 2020-02-26)
DCNv2\src\cpu\dcn_v2_cpu.cpp (2819, 2020-02-26)
DCNv2\src\cpu\vision.h (2665, 2020-02-26)
DCNv2\src\cuda (0, 2020-02-26)
DCNv2\src\cuda\dcn_v2_cuda.cu (20449, 2020-02-26)
DCNv2\src\cuda\dcn_v2_im2col_cuda.cu (20295, 2020-02-26)
DCNv2\src\cuda\dcn_v2_im2col_cuda.h (5226, 2020-02-26)
DCNv2\src\cuda\dcn_v2_psroi_pooling_cuda.cu (16214, 2020-02-26)
DCNv2\src\cuda\vision.h (2669, 2020-02-26)
DCNv2\src\dcn_v2.h (6176, 2020-02-26)
DCNv2\src\vision.cpp (891, 2020-02-26)
DCNv2\test.py (8506, 2020-02-26)
LICENSE (1520, 2020-02-26)
dcn_cpp_plugin (0, 2020-02-26)
dcn_cpp_plugin\CMakeLists.txt (1291, 2020-02-26)
dcn_cpp_plugin\dcn_v2.h (845, 2020-02-26)
dcn_cpp_plugin\dcn_v2_cuda.cu (6880, 2020-02-26)
dcn_cpp_plugin\dcn_v2_im2col_cuda.cu (8922, 2020-02-26)
dcn_cpp_plugin\dcn_v2_im2col_cuda.h (834, 2020-02-26)
dcn_cpp_plugin\vision.cpp (1436, 2020-02-26)
dcn_cpp_plugin\vision.h (726, 2020-02-26)
demo.py (691, 2020-02-26)
model.py (2373, 2020-02-26)
pose_dla_dcn.py (18236, 2020-02-26)
# DCN C PLUS CPLUS PLUGIN
## usage
void handle = dlopen("libdcn_v2_cuda_forward_v2.so", RTLD_LAZY);
int gpu_id = 0;
torch::jit::script::Module module =
torch::jit::load("centernet.pt", torch::Device(torch::DeviceType::CUDA, gpu_id));
近期下载者：
相关文件：
收藏者：