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.md
with additionaltorch_vision
package (MNIST dataset)update github action
test_codebase
andtest_tutorials
with additionaltorch_vision
packageadd
mnist_to_pointcloud
function indata/utils/utils.py
update
modules/data/load/loaders.py
with additionalmnist_to_pointcloud
functionupdate
modules/data/utils/custom_dataset
with specific repr for different dataset.add file
configs/dataset/MNIST.yaml
for MNIST datasetadd new plot_point_cloud function in
modules/utils/utils.py
for visualizationadd file
transforms/liftings/pointcloud2graph/knn_lifting.py
for KNN liftingupdate TRANSFORMS in
modules/transforms/data_transform.py
with additional GraphKNNLiftingadd file
modules/graph/gcn.py
add file
config/models/gcn.yaml
add example in
tutorial/pointcloud2graph/knn_lifiting.ipynb
Tutorial provided at notebook


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