Spaces:
Runtime error
Runtime error
Update kaggle_sam2_gpu_2.py
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
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 |
# 将提取的部分区域叠加到透明后的图片上
|