IFenton / scivision_huggingface_objectdetection

Hugging Face object detection plugin for scivision

Model repository for the scivision project that enables loading of object detection models from Hugging Face.

Via the scivision API, some of the top downloaded object detection models from Hugging Face (of models with a model card, last updated 31st May 2022) can be installed, loaded and run. The list of models is as follows:

1. facebook_detr_resnet_50
2. hustvl_yolos_small
3. detr_doc_table_detection
4. facebook_detr_resnet_101
5. hustvl_yolos_base
6. facebook_detr_resnet_50_dc5
7. hustvl_yolos_tiny

8. hustvl_yolos_small_300
9. hustvl_yolos_small_dwr

Models in this list can be loaded and used on data with a few lines of code, e.g.

from scivision import load_pretrained_model
this_repo = 'https://github.com/alan-turing-institute/scivision_huggingface_objectdetection'
model = load_pretrained_model(this_repo, model='facebook_detr_resnet_50')

You can then use the loaded model's predict function on image data loaded via scivision (see the user guide for details on how data is loaded via the scivision catalog):

model.predict(<image data>)