facebookresearch / dinov2

PyTorch code and models for the DINOv2 self-supervised learning method.
Apache License 2.0
8.32k stars 700 forks source link

Dinov2 Semantic Segmentation Notebook Fix #375

Open HarrisDePerceptron opened 4 months ago

HarrisDePerceptron commented 4 months ago

Broken dependencies for dinov2 segmentation

mmcv v2 and mmseg incompatiple with pytorch version on colab. i have create a pull request for the notebook please look into it.

Pull request: #370 Notebook

similar issues: #277
@Celebio @syhw @nlutsenko @colesbury @pdollar

tgolvano commented 4 months ago
Function Changes
init_segmentor Renamed to init_model
inference_segmentor Rename to inference_model
jenniferjtli commented 1 month ago

There are much more changes inside the codes. Backbone+head relatively easier to modify, while maskformer much more difficult because of missing Registry names (e.g Transformer sequence layer). Data transforms also need to be changed. Results nearly a hot mess, far from the demo website.