Permutohedral_attention_module
Permutohedral_attention_module copied to clipboard
Need some unit test
Unit test for the diferent components would be required on the longer run. Examples of such tests for the permutohedral lattice and the CRF can be found here: https://github.com/NifTK/NiftyNet/blob/dev/tests/crf_test.py As part of the c++ code in http://graphics.stanford.edu/projects/drf/ https://github.com/lucasb-eyer/pydensecrf/tree/master/tests https://github.com/MiguelMonteiro/permutohedral_lattice/tree/master/Tests https://github.com/MiguelMonteiro/CRFasRNNLayer/tree/master/Tests