/gradio_sample

Primary LanguagePython

gradio samples

This repo contains gradio demo codes of:

  • visualization of image preprocessing
  • image classification with ResNet50
  • semantic segmentation with DeepLabV3
  • object detection with YOLOX

All applications can be accessible on localhost:7860.

Usage

cd <PROJECT_ROOT>

# download yolox pre-trained weight
wget https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_s.onnx \
    -O weights/yolox_s.onnx

# build docker image
sh scripts/build_docker.sh

# run docker container
sh scripts/run_docker.sh

in docker environment

# visualize preprocess
python src/visualize_preprocess.py

# image classification
python src/image_classification.py

# semantic segmentation
python src/semantic_segmentation.py

# object detection
python src/object_detection.py

Reference