DrSleep / DenseTorch

An easy-to-use wrapper for work with dense per-pixel tasks in PyTorch (including multi-task learning)
MIT License
62 stars 9 forks source link

Evaluation scripts of multi-task tight-weight refineNet. #9

Closed fuxiao567 closed 4 years ago

fuxiao567 commented 4 years ago

would you like to share the evaluation scripts of multi-task tight-weight refinenet?Thank you.

DrSleep commented 4 years ago

you can run validation as in the example here: https://github.com/DrSleep/DenseTorch/blob/master/examples/multitask/train.py#L83-L90

fuxiao567 commented 4 years ago

Would you provide a more detailed test.py script to evaluate the test set? Thank you

DrSleep commented 4 years ago

what do you mean by "a more detailed script"? The provided example does both training and testing. If you comment out the training part (e.g, line 82), it will only run the testing part

fuxiao567 commented 4 years ago

Thank you for helping me solve this problem.