Code Monkey home page Code Monkey logo

tri-depth's People

Contributors

xingyuuchen avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

tri-depth's Issues

KeyError:('seg', 0, 0)

Hello, thank you very much for your contribution. I get the following error when adding triplet loss to another network: Traceback (most recent call last):
File "train.py", line 17, in
trainer.train()
File "/root/autodl-tmp/DIFFNet/trainer.py", line 179, in train
self.run_epoch()
File "/root/autodl-tmp/DIFFNet/trainer.py", line 195, in run_epoch
outputs, losses = self.process_batch(inputs)
File "/root/autodl-tmp/DIFFNet/trainer.py", line 253, in process_batch
losses = self.compute_losses(inputs, outputs)
File "/root/autodl-tmp/DIFFNet/trainer.py", line 508, in compute_losses
sgt_loss = self.compute_sgt_loss(inputs, outputs)
File "/root/autodl-tmp/DIFFNet/trainer.py", line 516, in compute_sgt_loss
seg_target = inputs[('seg', 0, 0)]
KeyError: ('seg', 0, 0)
There is no ('seg',0,0) key in inputs.How to solve such a problem? Looking forward to your response.

dataset problem

Your outstanding work is amazing. I tried to use your proposed loss for monodepth2, but there are always bugs when importing data, can you provide the code for monodepth2 about kitti_dataset and mono_datasets, thank you.

Original Traceback (most recent call last):
File "/home/yzhang/anaconda3/envs/mono/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop
data = fetcher.fetch(index)
File "/home/yzhang/anaconda3/envs/mono/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/yzhang/anaconda3/envs/mono/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/yzhang/monodepth2-master/datasets/kitti_dataset.py", line 176, in getitem
self.preprocess(inputs, color_aug)
File "/home/yzhang/monodepth2-master/datasets/kitti_dataset.py", line 121, in preprocess
self.a=color_aug(f)
TypeError: 'tuple' object is not callable

tri trainer

we find some error when we use your code to training our model,

‘seg_target = inputs[('seg', 0, 0)]‘show that there is no ’seg‘ in the inputs.

we notice that your code is based on FSRE and manydepth, so we copy the FSRE dataloader to tri-depth and debug the code, After making modifications to the data loader, we were able to successfully train our model using the provided code. We are curious to know if there are alternative methods for utilizing this code that we could explore

Pre-trained model

Thanks for releasing your amazing work. I am wondering if the pre-trained model is for ManyDepth?

Question about computation of total losses.

Hi, author. Thanks for your remarkable work.

loss += (reprojection_loss + consistency_loss) / (2 ** scale)
mean_disp = disp.mean(2, True).mean(3, True)
norm_disp = disp / (mean_disp + 1e-7)
smooth_loss = get_smooth_loss(norm_disp, color)
loss = loss + self.opt.disparity_smoothness * smooth_loss / (2 ** scale)
total_loss = total_loss + loss
losses["loss/{}".format(scale)] = loss
if not self.opt.disable_triplet_loss:
sgt_loss = self.compute_sgt_loss(inputs, outputs)
losses['sgt_loss'] = sgt_loss
total_loss = total_loss + sgt_loss * self.opt.sgt
losses["loss"] = total_loss

I noticed that the total_loss isn't divided by num_scales like original Manydepth does. Would this have an influence on the results? And I don't understand why smooth_loss and reprojection_loss are divided by 2**scale.
loss += (reprojection_loss + consistency_loss) / (2 ** scale)

loss = loss + self.opt.disparity_smoothness * smooth_loss / (2 ** scale)

colorful segmentation images

Dear author, the segmented picture provided in FSRE-Depth is gray, may I ask how the color segmented picture you provided in the article was obtained? I want to know how to get it, thank you!

Pretrained Model

Hello, it seems that the pre-trained model link cannot be downloaded

High-resolution results

Hi, thanks for your nice work!

I'm wondering have you tested the results of Tri-depth with high-resolution (320*1024) inputs? It would be helpful if you could provide the related information or results.

Looking forward to your reply!

Pretrained Models

Hi @xingyuuchen, thank you for open-sourcing your work!

I would like to ask if it is possible for you to release the pretrained weights of TriDepth on KITTI for the purpose of reproduction. Thanks!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.