fix msrvtt name from msvtt

ptpyip · Apr 10, 2024 · 636b360 · 636b360
1 parent 3bd0a2d
commit 636b360
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/data/msvtt_preprcoess.py → data/msrvtt_preprcoess.py b/data/msvtt_preprcoess.py → data/msrvtt_preprcoess.py
@@ -3,7 +3,7 @@
 import pandas as pd
 import  warnings 
 
-def msvtt_json2csv(json_path: str, csv_name = None):
+def msrvtt_json2csv(json_path: str, csv_name = None):
     assert json_path.endswith(".json")
     assert os.path.exists(json_path)
 
@@ -29,4 +29,4 @@ def msvtt_json2csv(json_path: str, csv_name = None):
     return
 
 if __name__ == "__main__":
-    msvtt_json2csv("./msrvtt_data/MSRVTT_data.json")
+    msrvtt_json2csv("./msrvtt_data/MSRVTT_data.json")
diff --git a/retrieval/dataloader/msvtt.py → retrieval/dataloader/msrvtt.py b/retrieval/dataloader/msvtt.py → retrieval/dataloader/msrvtt.py
@@ -10,7 +10,7 @@
 
 from .retrievalDataset import RetrievalDataset
 
-class MSVTT_Dataset(RetrievalDataset):
+class MSRVTTDataset(RetrievalDataset):
     """
     Dataset download: 
         - captions : https://github.com/ArrowLuo/CLIP4Clip/releases/download/v0.0/msrvtt_data.zip