Skip to content

preliminary version designed for depth estimation using Hist - for namla GPUs

Notifications You must be signed in to change notification settings

Chaouki-AI/DepthHistV01

Repository files navigation

DepthHist Model

Overview

This repository contains the DepthHist model, a preliminary version designed for depth estimation using a histogram-based approach. The model offers two configurations:

  • Simple Mode: A basic encoder-decoder architecture.
  • Non-Simple Mode: An advanced model that includes a histogram layer for enhanced depth computation.

Features

  • Encoder-Decoder Architecture: Leverages backbone models like EfficientNet or ResNet for feature extraction.
  • Histogram Layer: (Non-Simple Mode) Calculates depth by processing the encoder-decoder output through a histogram layer.
  • Flexible Backbone Support: Easily switch between different backbone architectures.

Installation

To get started, clone the repository and install the required dependencies:

git clone https://github.com/chaouki-ai/DepthHistV01
pip install -r requirements.txt

Usage

Here is a basic example of how to use the DepthHist model:

import torch
from Models.model import DepthHist

#  Example usage
model = DepthHist.build(bins=10, simple=False, backbone="efficientnet")
input_tensor = torch.randn(1, 3, 256, 256)  # Example input
output = model(input_tensor)

print(output.shape)  # Output depth map shape

Contact

This is a primer version of a model used for depth estimation using histogram approaches. For questions, suggestions, or collaborations, feel free to contact me:

Contributing

If you'd like to contribute to this project, please fork the repository and submit a pull request. Contributions are welcome!

License

This project is licensed under the MIT License. See the LICENSE file for more details.

About

preliminary version designed for depth estimation using Hist - for namla GPUs

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published