普林斯顿大学的团队出的新论文
github已开源,指路CornerNet-Lite github
本篇是实现记录:
依赖环境
- Python 3.7
- PyTorch 1.0.0
- CUDA 10
- GCC 4.9.2 or above
配置环境按照github上很好实现
测试下时间
本地环境
GTX1080
~/CornerNet-Lite$ python demo.py
demo.py 加了个时间输出
#!/usr/bin/env python
import cv2
import time
from core.detectors import CornerNet_Squeeze
from core.vis_utils import draw_bboxes
detect