目录
依赖项:
分割源代码:
依赖项:
groundingdino
Grounded-Segment-Anything
分割源代码:
generate_sky_mask.py
import os, sys
sys.path.append(os.getcwd())
# Change to your folder here
sys.path.append('Grounded-Segment-Anything')
import argparse
import os
import copy
import imageio
import numpy as np
import torch
from PIL import Image, ImageDraw, ImageFont
from torchvision.ops import box_convert
from huggingface_hub import hf_hub_download
from tqdm import tqdm
from termcolor import colored
from glob import glob
# Grounding DINO
import groundingdino.datasets.transforms as T
from groundingdino