evp_depth / evpconfig.py
MykolaL's picture
Upload EVPDepth
e1aaa1a verified
raw
history blame contribute delete
417 Bytes
from transformers import PretrainedConfig
class EVPConfig(PretrainedConfig):
model_type = "EVP"
def __init__(
self,
**kwargs,
):
self.num_deconv = 3
self.num_filters = [32,32,32]
self.deconv_kernels = [2,2,2]
self.dataset = 'nyudepthv2'
self.max_depth = 10
self.min_depth_eval = 1e-3
super().__init__(**kwargs)