Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
Hillobar authored Nov 20, 2023
1 parent c5c11fd commit a29d333
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions rope/VideoManager.py
Original file line number Diff line number Diff line change
Expand Up @@ -409,7 +409,6 @@ def process(self):

# Close video and process
if mfn['FrameNumber'] >= self.video_frame_total-1 or mfn['FrameNumber'] == self.stop_marker or self.play == False:
print ('here')
self.play_video("stop")
stop_time = float(self.capture.get(cv2.CAP_PROP_POS_FRAMES) / float(self.fps))
if stop_time == 0:
Expand Down Expand Up @@ -777,7 +776,7 @@ def apply_occlusion(self, img):
def apply_neg_CLIPs(self, img, CLIPText, CLIPAmount):
clip_mask = np.ones((352, 352))
CLIPimg = self.clip_transform(img).unsqueeze(0)

if CLIPText != "":
prompts = CLIPText.split(',')
with lock:
Expand All @@ -793,7 +792,8 @@ def apply_neg_CLIPs(self, img, CLIPText, CLIPAmount):
thresh = CLIPAmount/100.0
clip_mask[clip_mask>thresh] = 1.0
clip_mask[clip_mask<=thresh] = 0.0
return clip_mask
return clip_mask

# @profile
def apply_face_parser(self, img, FaceParserAmount):

Expand Down

0 comments on commit a29d333

Please sign in to comment.