Fix evaluate.py

echuraev · Jul 18, 2023 · b8b4886 · b8b4886
1 parent f831a40
commit b8b4886
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluate.py b/evaluate.py
@@ -1179,7 +1179,7 @@ def __init__(self, min_shape, preproc=None):
     def Validate(self, m, ref_outputs=[]):
         from tvm.contrib import download
         classes_url = "https://raw.githubusercontent.com/onnx/models/main/vision/object_detection_segmentation/faster-rcnn/dependencies/coco_classes.txt"
-        classes_fn = "coco_classes.txt"
+        classes_fn = "coco_classes_faster.txt"
         download.download(classes_url, classes_fn)
         classes = [line.rstrip('\n') for line in open(classes_fn)]