Skip to content

Add finetuning script and initial implementation#1

Merged
Devasy merged 2 commits intomainfrom
add-finetuning
Nov 17, 2024
Merged

Add finetuning script and initial implementation#1
Devasy merged 2 commits intomainfrom
add-finetuning

Conversation

@Devasy
Copy link
Copy Markdown
Owner

@Devasy Devasy commented Nov 17, 2024

Add initial implementation of DeepTurne package for fine-tuning CV-based deep models using Siamese architecture with triplet loss function.

  • README.md

    • Add project description, installation instructions, and usage examples.
  • deepturne/init.py

    • Initialize the deepturne package.
  • deepturne/triplet_loss.py

    • Implement custom triplet loss function.
  • deepturne/backbones.py

    • Add support for model backbones including ResNet, VGG, Inception, MobileNet, EfficientNet, Facenet, and Arcface.
    • Load pre-trained weights for Facenet and Arcface models.
  • deepturne/data_preprocessing.py

    • Implement data preprocessing steps including normalization, resizing, augmentation, and triplet creation.
  • deepturne/evaluation_metrics.py

    • Implement evaluation metrics including accuracy, precision, recall, F1 score, mean average precision (mAP), ROC-AUC, and triplet loss.
  • deepturne/datasets.py

    • Add support for downloading and loading basic datasets for fine-tuning, such as LFW, using TensorFlow Datasets (TFDS).
  • deepturne/test_finetuning.py

    • Implement a script to test the fine-tuning of at least one model for a few epochs.

For more details, open the Copilot Workspace session.

Add initial implementation of DeepTurne package for fine-tuning CV-based deep models using Siamese architecture with triplet loss function.

* **README.md**
  - Add project description, installation instructions, and usage examples.

* **deepturne/__init__.py**
  - Initialize the `deepturne` package.

* **deepturne/triplet_loss.py**
  - Implement custom triplet loss function.

* **deepturne/backbones.py**
  - Add support for model backbones including ResNet, VGG, Inception, MobileNet, EfficientNet, Facenet, and Arcface.
  - Load pre-trained weights for Facenet and Arcface models.

* **deepturne/data_preprocessing.py**
  - Implement data preprocessing steps including normalization, resizing, augmentation, and triplet creation.

* **deepturne/evaluation_metrics.py**
  - Implement evaluation metrics including accuracy, precision, recall, F1 score, mean average precision (mAP), ROC-AUC, and triplet loss.

* **deepturne/datasets.py**
  - Add support for downloading and loading basic datasets for fine-tuning, such as LFW, using TensorFlow Datasets (TFDS).

* **deepturne/test_finetuning.py**
  - Implement a script to test the fine-tuning of at least one model for a few epochs.

---

For more details, open the [Copilot Workspace session](https://copilot-workspace.githubnext.com/Devasy23/DeepTurne?shareId=XXXX-XXXX-XXXX-XXXX).
@Devasy Devasy merged commit b3e6473 into main Nov 17, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant