File size: 598 Bytes
7206ed3 6db6a87 63056c5 ea66b57 7206ed3 13575f0 6db6a87 56f79a3 36e49ec 56f79a3 13575f0 7206ed3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import os
import yaml
import numpy as np
from matplotlib import cm
import gradio as gr
import deeplabcut
import dlclibrary
import transformers
from PIL import Image
import requests
from deeplabcut.utils import auxiliaryfunctions
from dlclibrary.dlcmodelzoo.modelzoo_download import (
download_huggingface_model,
MODELOPTIONS,
)
model = 'superanimal_topviewmouse'
train_dir = 'DLC_models/sa-tvm'
download_huggingface_model(model, train_dir)
#demo data cooco cat:
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
image = Image.open(requests.get(url, stream=True).raw)
|