From 172bf1f1d617d49901a86c5c22fc31549a653295 Mon Sep 17 00:00:00 2001 From: Harry Date: Fri, 21 Jan 2022 03:12:45 +0800 Subject: [PATCH] Modify the convert.py file path and the anchor filename --- OneStage/yolo/deep_sort_yolov4/convert.py | 4 ++-- .../model_data/{yolo_anchors.txt => yolo4_anchors.txt} | 0 2 files changed, 2 insertions(+), 2 deletions(-) rename OneStage/yolo/deep_sort_yolov4/model_data/{yolo_anchors.txt => yolo4_anchors.txt} (100%) diff --git a/OneStage/yolo/deep_sort_yolov4/convert.py b/OneStage/yolo/deep_sort_yolov4/convert.py index ca93721..6855229 100644 --- a/OneStage/yolo/deep_sort_yolov4/convert.py +++ b/OneStage/yolo/deep_sort_yolov4/convert.py @@ -163,7 +163,7 @@ def close_session(self): model_path = 'yolo4_weight.h5' anchors_path = 'model_data/yolo4_anchors.txt' classes_path = 'model_data/coco_classes.txt' - weights_path = 'yolov4.weights' + weights_path = 'model_data/yolov4.weights' score = 0.5 iou = 0.5 @@ -172,4 +172,4 @@ def close_session(self): yolo4_model = Yolo4(score, iou, anchors_path, classes_path, model_path, weights_path) - yolo4_model.close_session() \ No newline at end of file + yolo4_model.close_session() diff --git a/OneStage/yolo/deep_sort_yolov4/model_data/yolo_anchors.txt b/OneStage/yolo/deep_sort_yolov4/model_data/yolo4_anchors.txt similarity index 100% rename from OneStage/yolo/deep_sort_yolov4/model_data/yolo_anchors.txt rename to OneStage/yolo/deep_sort_yolov4/model_data/yolo4_anchors.txt