mirror of
https://github.com/deepseek-ai/DreamCraft3D.git
synced 2025-02-23 14:28:55 -05:00
24 lines
503 B
Python
24 lines
503 B
Python
|
import random
|
||
|
|
from dataclasses import dataclass, field
|
||
|
|
|
||
|
|
import torch
|
||
|
|
import torch.nn as nn
|
||
|
|
import torch.nn.functional as F
|
||
|
|
|
||
|
|
import threestudio
|
||
|
|
from threestudio.utils.base import BaseModule
|
||
|
|
from threestudio.utils.typing import *
|
||
|
|
|
||
|
|
|
||
|
|
class BaseBackground(BaseModule):
|
||
|
|
@dataclass
|
||
|
|
class Config(BaseModule.Config):
|
||
|
|
pass
|
||
|
|
|
||
|
|
cfg: Config
|
||
|
|
|
||
|
|
def configure(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def forward(self, dirs: Float[Tensor, "B H W 3"]) -> Float[Tensor, "B H W Nc"]:
|
||
|
|
raise NotImplementedError
|