KNN lifting (Pointcloud to Graph) with New dataset MNIST #19
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Main changes:
With details:
update
readme.mdwith additionaltorch_visionpackage (MNIST dataset)update github action
test_codebaseandtest_tutorialswith additionaltorch_visionpackageadd
mnist_to_pointcloudfunction indata/utils/utils.pyupdate
modules/data/load/loaders.pywith additionalmnist_to_pointcloudfunctionupdate
modules/data/utils/custom_datasetwith specific repr for different dataset.add file
configs/dataset/MNIST.yamlfor MNIST datasetadd new plot_point_cloud function in
modules/utils/utils.pyfor visualizationadd file
transforms/liftings/pointcloud2graph/knn_lifting.pyfor KNN liftingupdate TRANSFORMS in
modules/transforms/data_transform.pywith additional GraphKNNLiftingadd file
modules/graph/gcn.pyadd file
config/models/gcn.yamladd example in
tutorial/pointcloud2graph/knn_lifiting.ipynbTutorial provided at notebook


from point cloud
to
by KNN lifting.
NOTE:
additional torch_vision package is required to get MNIST dataset.