bleakie / CenterMulti

基于CenterNet训练的目标检测&人脸对齐&姿态估计模型

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

如何获取TensorRT文件夹下程序依赖的代码

Boatsure opened this issue · comments

bleakie牛批!

我想将CenterNet通过TensorRT部署在Jetson TX2上,兜兜转转到了您这,想请问一下TensorRT文件夹底下的config、models.utils是来自哪个项目呢?CenterNet官方项目中没有这几样东西~

拜谢!

from config import cfg, update_config
from models.utils import _gather_feat, _transpose_and_gather_feat
from utils.image import get_affine_transform, transform_preds

find them! 来源于这个项目:https://github.com/tensorboy/centerpose