add pth

Signed-off-by: xujinling <jinling.xu@zilliz.com>

.idea/workspace.xml

@@ -0,0 +1,39 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ChangeListManager">
+    <list default="true" id="943b181f-2503-4903-b345-35ec6c2b2fd7" name="Changes" comment="" />
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="MarkdownSettingsMigration">
+    <option name="stateVersion" value="1" />
+  </component>
+  <component name="ProjectId" id="2AayTJJz1HmJfs1NKNpG7RTAIyY" />
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "RunOnceActivity.OpenProjectViewOnStart": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "last_opened_file_path": "/Users/zilliz/PycharmProjects/operator/video_text_embedding/frozen-in-time"
+  }
+}]]></component>
+  <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="Default task">
+      <changelist id="943b181f-2503-4903-b345-35ec6c2b2fd7" name="Changes" comment="" />
+      <created>1655261997162</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1655261997162</updated>
+    </task>
+    <servers />
+  </component>
+</project>

frozen_in_time.py

@@ -50,6 +50,8 @@ class FrozenInTime(NNOperator):
         super().__init__()
         self.model_name = model_name
         self.modality = modality
+        if weight_path is None:
+            weight_path = os.path.join(str(Path(__file__).parent), 'frozen_in_time_base_16_224.pth')
         if device is None:
             self.device = "cuda" if torch.cuda.is_available() else "cpu"
         else:
@@ -72,10 +74,10 @@ class FrozenInTime(NNOperator):
         self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False)
         self.transform_cfgs = get_configs(
             side_size=224,
-            crop_size=224,
+            crop_size=256,
             num_frames=self.num_frames,
-            mean=[0.48145466, 0.4578275, 0.40821073],
-            std=[0.26862954, 0.26130258, 0.27577711],
+            mean=[0.485, 0.456, 0.406],
+            std=[0.229, 0.224, 0.225],
         )
         self.model.eval()