Upload 6 files

data_info/__init__.py

@@ -0,0 +1,2 @@
+from .public_datasets import *
+from .pretrained_weight import *

data_info/pretrained_weight.py

@@ -0,0 +1,16 @@
+
+mldb_info={}
+
+mldb_info['checkpoint']={
+    'mldb_root': '/mnt/nas/share/home/xugk/ckpt', # NOTE: modify it to the pretrained ckpt root
+
+    # pretrained weight for convnext
+    'convnext_tiny': 'convnext/convnext_tiny_22k_1k_384.pth',
+    'convnext_small': 'convnext/convnext_small_22k_1k_384.pth',
+    'convnext_base': 'convnext/convnext_base_22k_1k_384.pth',
+    'convnext_large': 'convnext/convnext_large_22k_1k_384.pth',
+    'vit_large': 'vit/dinov2_vitl14_pretrain.pth',
+    'vit_small_reg': 'vit/dinov2_vits14_reg4_pretrain.pth',
+    'vit_large_reg': 'vit/dinov2_vitl14_reg4_pretrain.pth',
+    'vit_giant2_reg': 'vit/dinov2_vitg14_reg4_pretrain.pth',
+}

data_info/public_datasets.py

@@ -0,0 +1,7 @@
+mldb_info = {}
+
+mldb_info['NYU']={
+    'mldb_root': '/mnt/nas/share/home/xugk/data/',
+    'data_root': 'nyu',
+    'test_annotations_path': 'nyu/test_annotation.json',
+}

requirements.txt

@@ -0,0 +1,15 @@
+torch
+torchvision
+opencv-python
+numpy
+Pillow
+DateTime
+matplotlib
+plyfile
+HTML4Vision
+timm
+tensorboardX
+imgaug
+iopath
+imagecorruptions
+mmcv

requirements_v2.txt

@@ -0,0 +1,16 @@
+torch == 2.0.1
+torchvision == 0.15.2
+opencv-python
+numpy == 1.23.1
+xformers == 0.0.21.dev564
+Pillow
+DateTime
+matplotlib
+plyfile
+HTML4Vision
+timm
+tensorboardX
+imgaug
+iopath
+imagecorruptions
+mmcv