(This repo is deprecated now. Our new research is in DeepGHS.)
gchar_models
This is a repository for models used by gchar.
All the models' resourced will be listed below.
yolov5
All the models (with pattern of yolov5*.pt
) from ultralytics/yolov5 is placed at directory yolov5
.
You can download the model files with the following python code
import torch
from huggingface_hub import hf_hub_download
# attention to the side effect on matplotlib
# see: https://github.com/zhiqwang/yolov5-rt-stack/issues/467
from yolort.models import YOLOv5
model_file = hf_hub_download(
repo_id="narugo/gchar_models",
filename="yolov5/v6.0/yolov5s.pt"
)
model = YOLOv5.load_from_yolov5(model_file)
if torch.cuda.is_available():
model = model.cuda()
model.eval()