Update kaggle_sam2_gpu_1.py

kaggle_sam2_gpu_1.py

@@ -139,7 +139,10 @@ with sv.VideoSink(video_path, video_info=video_info) as sink:
 
         # 检查每个像素是否为白色，如果是白色，则保持其不透明
         #white_mask = (frame >= 100).all(axis=-1)
-        white_mask = (frame >= 130).all(axis=-1)
+        with open('/kaggle/white_mask_num.pkl', 'rb') as file:
+            white_mask_num = pickle.load(file)
+        print(white_mask_num)
+        white_mask = (frame >= white_mask_num).all(axis=-1)
         frame = np.where(white_mask[:, :, None], frame, transparent_frame)
 
         # 将提取的部分区域叠加到透明后的图片上