Update clip_model.py

clip_model.py

@@ -20,11 +20,27 @@ class ClipModel:
         self.processor = ChineseCLIPProcessor.from_pretrained(model_name)
 
         print("***** Clip Model LOAD DONE *****")
+        check_memory_usage()
 
         # Load Chinese vocabulary
         with open(vocab_path, 'r', encoding='utf-8') as f:
             self.vocab = [line.strip() for line in f.readlines()]
 
+    def check_memory_usage():
+        # Get memory details
+        memory_info = psutil.virtual_memory()
+    
+        total_memory = memory_info.total / (1024 * 1024)  # Convert bytes to MB
+        available_memory = memory_info.available / (1024 * 1024)
+        used_memory = memory_info.used / (1024 * 1024)
+        memory_usage_percent = memory_info.percent
+    
+        print(f"^^^^^^ Total Memory: {total_memory:.2f} MB ^^^^^^")
+        print(f"^^^^^^ Available Memory: {available_memory:.2f} MB ^^^^^^")
+        print(f"^^^^^^ Used Memory: {used_memory:.2f} MB ^^^^^^")
+        print(f"^^^^^^ Memory Usage (%): {memory_usage_percent}% ^^^^^^")
+
+
     def clip_result(self, image_path, top_k=3):
         """
         給定圖片路徑，返回最接近的 top_k 詞彙
@@ -33,6 +49,8 @@ class ClipModel:
         image = Image.open(image_path)
 
         print(f"===== Clip Model_clip_result : {image_path} ===== ")
+        # Run the function
+        check_memory_usage()
 
         # Process images and texts
         batch_size = 16  # Process 16 vocab at a time