|
|
|
|
|
|
|
|
|
import os |
|
import yaml |
|
import numpy as np |
|
from matplotlib import cm |
|
import gradio as gr |
|
import deeplabcut |
|
import dlclib |
|
|
|
from PIL import Image, ImageColor, ImageFont, ImageDraw |
|
|
|
from DLC_models.download_utils import DownloadModel |
|
from dlclive import DLCLive, Processor |
|
|
|
|
|
from viz_utils import save_results_as_json, draw_keypoints_on_image, draw_bbox_w_text, save_results_only_dlc |
|
from detection_utils import predict_md, crop_animal_detections, predict_dlc |
|
from ui_utils import gradio_inputs_for_MD_DLC, gradio_outputs_for_MD_DLC, gradio_description_and_examples |
|
|
|
|
|
|
|
import gradio as gr |
|
|
|
model = gr.Interface.load("mwmathis/DeepLabCutModelZoo-SuperAnimal-Quadruped", |
|
) |
|
|
|
model.launch() |
|
|
|
|
|
|