generated fromroboflow/template-python
- Notifications
You must be signed in to change notification settings - Fork1
DETR (resnet-50) module for use with Autodistill.
License
NotificationsYou must be signed in to change notification settings
autodistill/autodistill-detr
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
This repository contains the code supporting the DETR base model for use withAutodistill.
DETR is a transformer-based computer vision model you can use for object detection. Autodistill supports training a model using the Meta Research Resnet 50 checkpoint.
Read the fullAutodistill documentation.
Read theDETR Autodistill documentation.
To use DETR with autodistill, you need to install the following dependency:
pip3 install autodistill-detr
fromautodistill_detrimportDETR# load the modeltarget_model=DETR()# train for 10 epochstarget_model.train("./roads",epochs=10)# run inference on an imagetarget_model.predict("./roads/valid/-3-_jpg.rf.bee113a09b22282980c289842aedfc4a.jpg")
This project is licensed under anApache 2.0 license. See theHugging Face model card for the DETR Resnet 50 model for more information on the model license.
We love your input! Please see the core Autodistillcontributing guide to get started. Thank you 🙏 to all our contributors!