(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()