Upload 6 files
Browse files- data_info/__init__.py +2 -0
- data_info/pretrained_weight.py +16 -0
- data_info/public_datasets.py +7 -0
- requirements.txt +15 -0
- requirements_v2.txt +16 -0
- weight/新建文本文档.txt +0 -0
data_info/__init__.py
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from .public_datasets import *
|
| 2 |
+
from .pretrained_weight import *
|
data_info/pretrained_weight.py
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
mldb_info={}
|
| 3 |
+
|
| 4 |
+
mldb_info['checkpoint']={
|
| 5 |
+
'mldb_root': '/mnt/nas/share/home/xugk/ckpt', # NOTE: modify it to the pretrained ckpt root
|
| 6 |
+
|
| 7 |
+
# pretrained weight for convnext
|
| 8 |
+
'convnext_tiny': 'convnext/convnext_tiny_22k_1k_384.pth',
|
| 9 |
+
'convnext_small': 'convnext/convnext_small_22k_1k_384.pth',
|
| 10 |
+
'convnext_base': 'convnext/convnext_base_22k_1k_384.pth',
|
| 11 |
+
'convnext_large': 'convnext/convnext_large_22k_1k_384.pth',
|
| 12 |
+
'vit_large': 'vit/dinov2_vitl14_pretrain.pth',
|
| 13 |
+
'vit_small_reg': 'vit/dinov2_vits14_reg4_pretrain.pth',
|
| 14 |
+
'vit_large_reg': 'vit/dinov2_vitl14_reg4_pretrain.pth',
|
| 15 |
+
'vit_giant2_reg': 'vit/dinov2_vitg14_reg4_pretrain.pth',
|
| 16 |
+
}
|
data_info/public_datasets.py
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
mldb_info = {}
|
| 2 |
+
|
| 3 |
+
mldb_info['NYU']={
|
| 4 |
+
'mldb_root': '/mnt/nas/share/home/xugk/data/',
|
| 5 |
+
'data_root': 'nyu',
|
| 6 |
+
'test_annotations_path': 'nyu/test_annotation.json',
|
| 7 |
+
}
|
requirements.txt
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch
|
| 2 |
+
torchvision
|
| 3 |
+
opencv-python
|
| 4 |
+
numpy
|
| 5 |
+
Pillow
|
| 6 |
+
DateTime
|
| 7 |
+
matplotlib
|
| 8 |
+
plyfile
|
| 9 |
+
HTML4Vision
|
| 10 |
+
timm
|
| 11 |
+
tensorboardX
|
| 12 |
+
imgaug
|
| 13 |
+
iopath
|
| 14 |
+
imagecorruptions
|
| 15 |
+
mmcv
|
requirements_v2.txt
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch == 2.0.1
|
| 2 |
+
torchvision == 0.15.2
|
| 3 |
+
opencv-python
|
| 4 |
+
numpy == 1.23.1
|
| 5 |
+
xformers == 0.0.21.dev564
|
| 6 |
+
Pillow
|
| 7 |
+
DateTime
|
| 8 |
+
matplotlib
|
| 9 |
+
plyfile
|
| 10 |
+
HTML4Vision
|
| 11 |
+
timm
|
| 12 |
+
tensorboardX
|
| 13 |
+
imgaug
|
| 14 |
+
iopath
|
| 15 |
+
imagecorruptions
|
| 16 |
+
mmcv
|
weight/新建文本文档.txt
ADDED
|
File without changes
|