Adding some lines of code for CNN backbone for Research

[Shalinis19137]

Description

This update integrates a Convolutional Neural Network (CNN) backbone to enhance feature extraction capabilities for our research project. This addition aims to improve model performance on image datasets, addressing the limitations of the existing architecture.

Fixes #61
Dependencies

• TensorFlow >= 2.0 or PyTorch >= 1.6

• Keras or torchvision
  Type of change

• New feature (non-breaking change which adds functionality)

How Has This Been Tested?

• Test A : Verified integration with synthetic image datasets.
• Test B : Benchmarked performance on real-world datasets.

Instructions:

1. Test with a small dataset to confirm CNN functionality.
2. Scale up to more complex datasets for performance evaluation.

Checklist:

• Code adheres to project style guidelines.
• Self-reviewed and commented code.
• Updated documentation.
• No new warnings generated.
• Added tests for the CNN backbone.
• All tests pass locally.
• Dependencies merged and published.

I am a beginner in this field if you like my work please provide points under gssocs-ext 2024

[Shalinis19137]

I already resolved this problem in #80 repo.

[Devasy]

@Shalinis19137 this changes shouldn't go with the main.py file, for that we have model-training branch

[Devasy]

@Shalinis19137 this changes shouldn't go with the main.py file, for that we have model-training branch

@Shalinis19137 any update ?

[Shalinis19137]

Still working on it sir