search

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

Get Dataset's Prediction #756

Closed fangwudi closed 3 years ago

fangwudi commented 3 years ago

❓ Questions and Help

Is there any easy way to get total prediction of one dataset's testset? Do not need meters, only prediction is needed.

mannatsingh commented 3 years ago

Do you mean getting all the predictions for a dataset? You can use https://github.com/facebookresearch/ClassyVision/blob/master/classy_vision/hooks/output_csv_hook.py for that. If you just want to run inference, you can call set_test_only to True.

If you just want aggregate counts, then you can simply write a custom meter which reports this :)

fangwudi commented 3 years ago

Do you mean getting all the predictions for a dataset? You can use https://github.com/facebookresearch/ClassyVision/blob/master/classy_vision/hooks/output_csv_hook.py for that. If you just want to run inference, you can call set_test_only to True.

If you just want aggregate counts, then you can simply write a custom meter which reports this :)

thanks. I think output_csv_hook maybe helpful.