search

longzw1997 / Open-GroundingDino

This is the third party implementation of the paper Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection.
MIT License
457 stars 71 forks source link

Added A example Training Script #77

Closed SahilCarterr closed 5 months ago

SahilCarterr commented 5 months ago

Pull Request Title:

Added an Example Training Script Regarding Issues #75 and #72

Description:

This pull request addresses issues #75 and #72 by adding an example training script to the repository. The script demonstrates how to train model using the custom dataset, offering users a clear and practical example to follow.

Changes Made:

  1. New Script Added: examples/Training_Script_example.ipynb
    • This script includes detailed comments and explanations for each step in the training process.
    • It covers data loading, model creation, training loop, and evaluation.
    • Includes configurations for common training parameters.

Additional Notes:

Feel free to suggest any improvements or additional features that could be included in this example script. Your feedback is highly appreciated!

Please review the changes and let me know if there are any adjustments needed. Thank you!

EduardoPach commented 5 months ago

@SahilCarterr hey, thanks for sharing the notebook it would be great if you could also share a minimal dataset (something like 10 samples) to be able to actually run the whole thing

SahilCarterr commented 5 months ago

Hey @EduardoPach, create an issue, then I will create a pull request from the same