hysts HF staff commited on
Commit
dc16691
1 Parent(s): 32dc906
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -586,12 +586,14 @@ def preprocess_image_end(image_end):
586
 
587
 
588
  def add_drag(tracking_points):
589
- tracking_points.append([])
 
590
  return tracking_points
591
 
592
 
593
  def delete_last_drag(tracking_points, first_frame_path, last_frame_path):
594
- tracking_points.pop()
 
595
  transparent_background = Image.open(first_frame_path).convert("RGBA")
596
  transparent_background_end = Image.open(last_frame_path).convert("RGBA")
597
  w, h = transparent_background.size
@@ -633,13 +635,16 @@ def delete_last_drag(tracking_points, first_frame_path, last_frame_path):
633
 
634
 
635
  def delete_last_step(tracking_points, first_frame_path, last_frame_path):
636
- tracking_points[-1].pop()
 
637
  transparent_background = Image.open(first_frame_path).convert("RGBA")
638
  transparent_background_end = Image.open(last_frame_path).convert("RGBA")
639
  w, h = transparent_background.size
640
  transparent_layer = np.zeros((h, w, 4))
641
 
642
  for track in tracking_points:
 
 
643
  if len(track) > 1:
644
  for i in range(len(track) - 1):
645
  start_point = track[i]
 
586
 
587
 
588
  def add_drag(tracking_points):
589
+ if not tracking_points or tracking_points[-1]:
590
+ tracking_points.append([])
591
  return tracking_points
592
 
593
 
594
  def delete_last_drag(tracking_points, first_frame_path, last_frame_path):
595
+ if tracking_points:
596
+ tracking_points.pop()
597
  transparent_background = Image.open(first_frame_path).convert("RGBA")
598
  transparent_background_end = Image.open(last_frame_path).convert("RGBA")
599
  w, h = transparent_background.size
 
635
 
636
 
637
  def delete_last_step(tracking_points, first_frame_path, last_frame_path):
638
+ if tracking_points and tracking_points[-1]:
639
+ tracking_points[-1].pop()
640
  transparent_background = Image.open(first_frame_path).convert("RGBA")
641
  transparent_background_end = Image.open(last_frame_path).convert("RGBA")
642
  w, h = transparent_background.size
643
  transparent_layer = np.zeros((h, w, 4))
644
 
645
  for track in tracking_points:
646
+ if not track:
647
+ continue
648
  if len(track) > 1:
649
  for i in range(len(track) - 1):
650
  start_point = track[i]