diff --git a/README.md b/README.md index d004d9f23ccb10262833d02d48c554af139a2e9b..40883fc9077f0261f8b89213da80293361ea03e6 100644 --- a/README.md +++ b/README.md @@ -99,20 +99,23 @@ h = h*dh ## 预训练模型 -### 1、手部检测预训练模型 +### 1、COCO 预训练模型 +* [预训练模型下载地址(百度网盘 Password: egdg )](https://pan.baidu.com/s/1KTvktMA38lsVYUjQ0sh5HQ) + +### 2、手部检测预训练模型 * 包括yolo预训练模型图像输入尺寸416,yolo-tiny预训练模型图像输入尺寸为512。 * [预训练模型下载地址(百度网盘 Password: 7mk0 )](https://pan.baidu.com/s/1hqzvz0MeFX0EdpWXUV6aFg) -### 2、脸部检测预训练模型 +### 3、脸部检测预训练模型 * [预训练模型下载地址(百度网盘 Password: l2a3 )](https://pan.baidu.com/s/1xVtZUMD94DiT9FQQ66xG1A) -### 3、人检测预训练模型 +### 4、人检测预训练模型 * [预训练模型下载地址(百度网盘 Password: ise9 )](https://pan.baidu.com/s/1mxiI-tOpE3sU-9TVPJmPWw) -### 4、安全帽检测预训练模型 +### 5、安全帽检测预训练模型 * [预训练模型下载地址(百度网盘 Password: inu8 )](https://pan.baidu.com/s/1rWUEsPnOTdmfJW4xM8m3Eg) -### 5、交通工具预训练模型 +### 6、交通工具预训练模型 * [预训练模型下载地址(百度网盘 Password: 9qlx )](https://pan.baidu.com/s/1IwTnQYulFetY-AS1woMMaQ) ## 项目使用方法 diff --git a/cfg/coco.data b/cfg/coco.data new file mode 100644 index 0000000000000000000000000000000000000000..193c946b9c8179cdf8552b2b8a97b10ed19d5494 --- /dev/null +++ b/cfg/coco.data @@ -0,0 +1,13 @@ +cfg_model=yolo +classes=80 +gpus = 0 +num_workers = 8 +batch_size = 16 +img_size = 416 +multi_scale = True +epochs = 100 +train=./datasets_train/anno/train.txt +valid=./datasets_train/anno/train.txt +names=./cfg/coco.names +finetune_model=./coco_model/yolov3_coco.pt +lr_step = 30,40,60 diff --git a/cfg/coco.names b/cfg/coco.names new file mode 100644 index 0000000000000000000000000000000000000000..941cb4e1392266f6a6c09b1fdc5f79503b2e5df6 --- /dev/null +++ b/cfg/coco.names @@ -0,0 +1,80 @@ +person +bicycle +car +motorcycle +airplane +bus +train +truck +boat +traffic light +fire hydrant +stop sign +parking meter +bench +bird +cat +dog +horse +sheep +cow +elephant +bear +zebra +giraffe +backpack +umbrella +handbag +tie +suitcase +frisbee +skis +snowboard +sports ball +kite +baseball bat +baseball glove +skateboard +surfboard +tennis racket +bottle +wine glass +cup +fork +knife +spoon +bowl +banana +apple +sandwich +orange +broccoli +carrot +hot dog +pizza +donut +cake +chair +couch +potted plant +bed +dining table +toilet +tv +laptop +mouse +remote +keyboard +cell phone +microwave +oven +toaster +sink +refrigerator +book +clock +vase +scissors +teddy bear +hair drier +toothbrush diff --git a/cfg/coco_mini.data b/cfg/coco_mini.data new file mode 100644 index 0000000000000000000000000000000000000000..efe72384e0ef593fe6d66efa5b850e53bc8bce54 --- /dev/null +++ b/cfg/coco_mini.data @@ -0,0 +1,15 @@ +cfg_model=yolo +classes=6 +gpus = 0 +num_workers = 8 +batch_size = 8 +img_size = 416 +multi_scale = True +epochs = 100 +train=./datasets_coco_mini/anno/train.txt +valid=./datasets_coco_mini/anno/train.txt +names=./cfg/coco_mini.names +finetune_model=./coco_model/yolov3_coco.pt +#finetune_model = ./weights-yolov3-coco_mini/latest_416.pt +lr_step = 10,20,30 +lr0 = 0.001 diff --git a/cfg/coco_mini.names b/cfg/coco_mini.names new file mode 100644 index 0000000000000000000000000000000000000000..84b0e4e41da8fcc44eac6394f7081f5c5ea401de --- /dev/null +++ b/cfg/coco_mini.names @@ -0,0 +1,6 @@ +person +car +motorcycle +dog +bus +truck