Tob-iee/detr_model_loader.py

Created September 13, 2023 14:19

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/Tob-iee/8c5fe643d5b2d8bce8cfb8802e559acc.js"></script>
Save Tob-iee/8c5fe643d5b2d8bce8cfb8802e559acc to your computer and use it in GitHub Desktop.

Download ZIP

Load the image processor (encoder) and object detector (decoder).

Raw

detr_model_loader.py

	image_processor = AutoImageProcessor.from_pretrained(finetuned_output_dir)
	model = AutoModelForObjectDetection.from_pretrained(finetuned_output_dir)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment