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()