From e5a3e0ba22733e0bf3c224319a630093c73a70e1 Mon Sep 17 00:00:00 2001 From: Satyam Bharti <44036099+satyambharti171@users.noreply.github.com> Date: Tue, 7 Jul 2020 17:21:54 +0530 Subject: [PATCH] Modified convert.py with more suitable locations --- OneStage/yolo/deep_sort_yolov4/convert.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/OneStage/yolo/deep_sort_yolov4/convert.py b/OneStage/yolo/deep_sort_yolov4/convert.py index ca93721..e8429d6 100644 --- a/OneStage/yolo/deep_sort_yolov4/convert.py +++ b/OneStage/yolo/deep_sort_yolov4/convert.py @@ -161,9 +161,9 @@ def close_session(self): if __name__ == '__main__': model_path = 'yolo4_weight.h5' - anchors_path = 'model_data/yolo4_anchors.txt' + anchors_path = 'model_data/yolo_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()