centernet
古风子
lalala
展开
-
centernet-训练和验证自己的数据集(安全帽)
本文以安全帽检测为例,验证下训练自己的数据集1. 创建json数据将自己的coco类型的数据放到根目录的dataset目录下~/AI/CenterNet-master/data/helmet$ lsannotations images kmeans.py meanvalue.txt/annotations$ lsinstances_test2019.json instances_train2019.json instances_val2019.jsonannotation原创 2020-07-15 21:12:23 · 2617 阅读 · 13 评论 -
centernet-环境搭建
1. 创建pytorch环境在已有的pytorch虚拟环境上复制环境conda create -n certernet 这是使用pytorch1.2和CUDA 10.0;多CUDA共存请参考服务器多cuda共存因为网络在线安装,容易断连,因此可以下载离线安装包离线pytorch1.2下载地址conda install --offline pytorch-1.2.0-py3.6_cuda10.0.130_cudnn7.6.2_0.tar.bz2#安装pytorch依赖的其他包conda i原创 2020-07-15 21:08:40 · 1374 阅读 · 0 评论 -
centernet-检测过程解析(安全帽场景)
检测python demo.py ctdet --demo ../data//helmet/test2019/07022055230002942.jpg --arch res_18 --load_model ../exp/ctdet/helmet/model_best.pth ctdet表示task名称,表示普通的物体检测功能,对应的实际功能类为CtdetDetector.py–demo 表示要处理的数据,或者数据目录–arch 表示检测使用的backbone网络类型–load_model原创 2020-08-25 17:53:46 · 2912 阅读 · 8 评论