Training and testing issue

brandleyzhou / DIFFNet

[BMVC 2021] ''Self-Supervised Monocular Depth Estimation with Internal Feature Fusion''

109 stars 21 forks source link

Training and testing issue #17

Closed ArminMasoumian closed 2 years ago

ArminMasoumian commented 2 years ago

Hi,

When I'm trying to test a simple image by running the " sh test_sample.sh " code, I get this error: " ModuleNotFoundError: No module named 'hr_networks' "

Would you please let me know how I can get "hr_networks"?

Also when I tried to train the model, this error popup:

from .hrnet_config import MODEL_CONFIGS File "/media/armin/DATA/DIFFNet/networks/hrnet_config.py", line 5, in from yacs.config import CfgNode as CN ModuleNotFoundError: No module named 'yacs'

Do I miss something here?

brandleyzhou commented 2 years ago

you can delete 'import hr_networks' in test_sample.py
for problem 2 : you can install the required lib by "pip install yacs"

ArminMasoumian commented 2 years ago

you can delete 'import hr_networks' in test_sample.py

for problem 2 : you can install the required lib by "pip install yacs"

Tnx, the provided code gives the results for 640x192 image size. where can I change it to the original size input (1024x320) and train with that? Also, it seems that you add an internal feature fusion to the original HRNet, I would like to remove that and test it with the original HRNet. In the "test_hr_encoder.py" I tried to remove "mixed_features" and only return "features", but in the decoder, I get an error. Is there any way to train your model with the original HRNet?