Mirrors/Track-Anything
1
0
Fork 0
mirror of https://github.com/gaomingqi/Track-Anything.git synced 2025-12-16 16:37:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix the template color

Browse Source
This commit is contained in:
Yioutpi
2023-04-14 15:01:11 +08:00
parent 2809c45402
commit d84efee187
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app.py
View File

@@ -41,7 +41,7 @@ xmem_checkpoint = download_checkpoint(xmem_checkpoint_url, folder, xmem_checkpoi
# args, defined in track_anything.py # args, defined in track_anything.py
args = parse_augment() args = parse_augment()
args.port=12212 args.port=12219
model = TrackingAnything(SAM_checkpoint, xmem_checkpoint, args) model = TrackingAnything(SAM_checkpoint, xmem_checkpoint, args)
@@ -99,8 +99,8 @@ def get_frames_from_video(video_input, play_state):
except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e: except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
print("read_frame_source:{} error. {}\n".format(video_path, str(e))) print("read_frame_source:{} error. {}\n".format(video_path, str(e)))
for frame in frames: for index, frame in enumerate(frames):
frame = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)) frames[index] = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
key_frame_index = int(timestamp * fps) key_frame_index = int(timestamp * fps)
nearest_frame = frames[key_frame_index] nearest_frame = frames[key_frame_index]