Skip to content

Dino model#74

Open
OmarJabri7 wants to merge 9 commits intoivy-llc:mainfrom
OmarJabri7:dino_model
Open

Dino model#74
OmarJabri7 wants to merge 9 commits intoivy-llc:mainfrom
OmarJabri7:dino_model

Conversation

@OmarJabri7
Copy link

Finalized DINO model architecture split into three main files:

  • dino.py containing the DINONet and Spec classes
  • layers.py containing the inner layers that the DINI model builds on top: DINO Back Bone ( vision Transfomer) and DINO head (projection head)
  • utils.py containing utility functions that the different DINO modules use

and a testing script under test_dinonet.py, which displays the DINO model in action against image data.

@a0m0rajab
Copy link

@OmarJabri7 That's a great work you did, can you message me when this is merged so that we can have a blog post draft about it?

… expand_dims \nNeed to fix DINO net with nlayers > 1, currently nlayers are 1 and it is working fine
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.

2 participants