Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add bts_sequence for pytorch #45

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Conversation

kHarshit
Copy link
Contributor

@kHarshit kHarshit commented Feb 13, 2020

  • Added bts_sequence.py for testing multiple images in a folder.

Note: Currently it's not using DataLoader.


Working result:

(pylatest) [kharshit@temp28 pytorch]$ CUDA_VISIBLE_DEVICES=0 python bts_sequence.py --image_path /nfs/interns/kharshit/DepthEstimation/bts/dataset/test_images/
Testing 17 files with ../models/bts_nyu_v2_pytorch_densenet161/model
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 17/17 [00:01<00:00,  9.73it/s]
Elapesed time: 1.7538671493530273
Done.
Saving result pngs..
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 17/17 [00:00<00:00, 29.96it/s]

@rsj007
Copy link

rsj007 commented Dec 2, 2021

Hello, I am running your bts_sequence.py with eigen model on an image captured by a camera. When I run it, it shows error as below. I don't know where to import this. Thanks a lot!!
image

@kHarshit
Copy link
Contributor Author

kHarshit commented Dec 2, 2021

@rsj007 you need to provide the model path with command line argument --model_name <model path>

@rsj007
Copy link

rsj007 commented Dec 2, 2021

@rsj007 you need to provide the model path with command line argument --model_name <model path>

Thanks. I solved it.
Can I test an image with size 1920*1200?

@kHarshit
Copy link
Contributor Author

kHarshit commented Dec 2, 2021

As far as I remember, your image size should be a multiple of 32. Based on the GPU RAM, you can try testing it.

@rsj007
Copy link

rsj007 commented Dec 2, 2021

As far as I remember, your image size should be a multiple of 32. Based on the GPU RAM, you can try testing it.

Ok! I noticed that the height of an image will be divised by 16,8,4. Maybe a multiple if 16?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants