Merge pull request #6 from ShuoShenDe/main

feat: create grounded_sam2_tracking_demo_with_continuous_id.py
This commit is contained in:
Ren Tianhe
2024-08-09 09:17:01 +08:00
committed by GitHub
187 changed files with 458 additions and 13 deletions

View File

@@ -3,7 +3,7 @@ import torch
import numpy as np
import supervision as sv
from supervision.draw.color import ColorPalette
from supervision_utils import CUSTOM_COLOR_MAP
from utils.supervision_utils import CUSTOM_COLOR_MAP
from PIL import Image
from sam2.build_sam import build_sam2
from sam2.sam2_image_predictor import SAM2ImagePredictor