supersolar commited on
Commit
bdb5bf4
·
verified ·
1 Parent(s): 6da1a91

Update kaggle_sam2_gpu_2.py

Browse files
Files changed (1) hide show
  1. kaggle_sam2_gpu_2.py +6 -1
kaggle_sam2_gpu_2.py CHANGED
@@ -139,7 +139,12 @@ with sv.VideoSink(video_path, video_info=video_info) as sink:
139
 
140
  # 检查每个像素是否为白色,如果是白色,则保持其不透明
141
  #white_mask = (frame >= 100).all(axis=-1)
142
- white_mask = (frame >= 130).all(axis=-1)
 
 
 
 
 
143
  frame = np.where(white_mask[:, :, None], frame, transparent_frame)
144
 
145
  # 将提取的部分区域叠加到透明后的图片上
 
139
 
140
  # 检查每个像素是否为白色,如果是白色,则保持其不透明
141
  #white_mask = (frame >= 100).all(axis=-1)
142
+
143
+ with open('/kaggle/white_mask_num.pkl', 'rb') as file:
144
+ white_mask_num = pickle.load(file)
145
+ print(white_mask_num)
146
+ white_mask = (frame >= white_mask_num).all(axis=-1)
147
+ #white_mask = (frame >= 130).all(axis=-1)
148
  frame = np.where(white_mask[:, :, None], frame, transparent_frame)
149
 
150
  # 将提取的部分区域叠加到透明后的图片上