search

roboflow / notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.
https://roboflow.com/models
4.89k stars 759 forks source link

Add DINOv2 image retrieval notebook #133

Closed capjamesg closed 1 year ago

capjamesg commented 1 year ago

Description

This PR includes a notebook showing how to do image retrieval with DINOv2 and faiss.

SkalskiP commented 1 year ago

Hi @capjamesg 👋🏻 Same story here. It Looks like you started from an outdated version of the notebooks repository. Please resolve the conflicts in README.md and automation/notebooks-table-data.csv. 🙏🏻

SkalskiP commented 1 year ago

It is hard to leave comments on notebooks, so I'm going to post stuff here.

  1. Let's use the new Roboflow Universe login. Brad wanted us to promote it.
import roboflow
from roboflow import Roboflow

roboflow.login()

rf = Roboflow()
project = rf.workspace("team-roboflow").project("coco-128")
dataset = project.version(2).download("coco")
  1. You use from google.colab.patches import cv2_imshow. This will not work on Kaggle or SegeMaker Studio Labs. I suggest using different ways of plotting the image.
capjamesg commented 1 year ago

@SkalskiP I have incorporated all of your feedback into the notebook.

SkalskiP commented 1 year ago

It looks like we are missing pip install supervision command.

SkalskiP commented 1 year ago

Also, it looks like after https://github.com/roboflow/notebooks/pull/132 merge, we need to resolve conflicts.

capjamesg commented 1 year ago

pip install added and merge conflicts fixed!