Skip to content

secortot/detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

kaggle-open-image2019

centernet已经可以直接运行测试,按照centernet的README下载相应的预训练模型到centernet/models文件夹(https://github.com/xingyizhou/CenterNet);按照网页链接的步骤来

在ctdet.py文件下show_results函数下修改结果的存储路径;

环境:RTX2080ti、Ubuntu18.04、pytorch1.1、cuda10.0

注意:如果使用的是pytorch1.x,因为1.x移除了ffi,则/src/models/DCNV2/src下的build.py和build_double.py文件需要修改

from torch.utils.ffi import create_extension替换为from torch.utils.cpp_extension import BuildExtension

ffi = create_extension替换为ffi = BuildExtension

在Cornernet目录下运行python demo.py ctdet --demo /path/to/image/or/folder/or/video --load_model ../models/ctdet_coco_dla_2x.pth

About

kaggle-open-image2019

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published