feat(vla): vla框架初始化
This commit is contained in:
6
roboimi/vla/models/backbones/__init__.py
Normal file
6
roboimi/vla/models/backbones/__init__.py
Normal file
@@ -0,0 +1,6 @@
|
||||
# Backbone models
|
||||
from .siglip import SigLIPBackbone
|
||||
from .clip import CLIPBackbone
|
||||
from .dinov2 import DinoV2Backbone
|
||||
|
||||
__all__ = ["SigLIPBackbone", "CLIPBackbone", "DinoV2Backbone"]
|
||||
1
roboimi/vla/models/backbones/clip.py
Normal file
1
roboimi/vla/models/backbones/clip.py
Normal file
@@ -0,0 +1 @@
|
||||
# CLIP Backbone 实现
|
||||
1
roboimi/vla/models/backbones/dinov2.py
Normal file
1
roboimi/vla/models/backbones/dinov2.py
Normal file
@@ -0,0 +1 @@
|
||||
# DinoV2 Backbone 实现
|
||||
1
roboimi/vla/models/backbones/siglip.py
Normal file
1
roboimi/vla/models/backbones/siglip.py
Normal file
@@ -0,0 +1 @@
|
||||
# SigLIP Backbone 实现
|
||||
5
roboimi/vla/models/heads/__init__.py
Normal file
5
roboimi/vla/models/heads/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
# Action Head models
|
||||
from .diffusion import DiffusionActionHead
|
||||
from .act import ACTHead
|
||||
|
||||
__all__ = ["DiffusionActionHead", "ACTHead"]
|
||||
1
roboimi/vla/models/heads/act.py
Normal file
1
roboimi/vla/models/heads/act.py
Normal file
@@ -0,0 +1 @@
|
||||
# ACT-VAE Action Head 实现
|
||||
1
roboimi/vla/models/heads/diffusion.py
Normal file
1
roboimi/vla/models/heads/diffusion.py
Normal file
@@ -0,0 +1 @@
|
||||
# Diffusion Policy Action Head 实现
|
||||
5
roboimi/vla/models/projectors/__init__.py
Normal file
5
roboimi/vla/models/projectors/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
# Projector models
|
||||
from .mlp import MLPProjector
|
||||
from .perceiver import PerceiverResampler
|
||||
|
||||
__all__ = ["MLPProjector", "PerceiverResampler"]
|
||||
1
roboimi/vla/models/projectors/mlp.py
Normal file
1
roboimi/vla/models/projectors/mlp.py
Normal file
@@ -0,0 +1 @@
|
||||
# MLP Projector 实现
|
||||
1
roboimi/vla/models/projectors/perceiver.py
Normal file
1
roboimi/vla/models/projectors/perceiver.py
Normal file
@@ -0,0 +1 @@
|
||||
# Perceiver Resampler 实现
|
||||
Reference in New Issue
Block a user