|
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) |
|
|
|
|
|
url = "http://images.cocodataset.org/val2017/000000039769.jpg" |
|
image = Image.open(requests.get(url, stream=True).raw) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
