lannio/Point-GNN.pytorchPublic forked fromdatong-new/Point-GNN.pytorch NotificationsYou must be signed in to change notification settings Fork0 Star0 Code Pull requests Actions Projects Security Insights Additional navigation options Files master configs dataset models scripts splits util .gitignore .gitmodules LICENSE READ...
Breadcrumbs Point-GNN.pytorch / run.pyTop File metadata and controls Code Blame 540 lines (535 loc) · 26.6 KB Raw """This file implement an inference pipeline for Point-GNN on KITTI dataset""" import os import time import argparse import multiprocessing from functools import partial import ...
mamba install pytorch torchvision sage=10.0 pytorch-cuda=11.7 pyg=2.3 pytorch-scatter pytorch-sparse pytorch-cluster torchmetrics rtree -c pyg -c pytorch -c nvidia -c conda-forge pip install abspy==0.2.6 hydra-core hydra-colorlog omegaconf trimesh tqdm wandb plyfile 🚀 Usage Quick start Do...
NOTE: installing from the .yaml file may cause pytorch to install incorrectly (CPU version instead of GPU version). Alternate installation instructions: # create and activate environment mamba create -n thermoMPNN python=3.10 mamba activate thermoMPNN # install pytorch related packages mamba install ...
The focus is on a modular implementation with the possibility to incorporate different invariances and equivariances into the overall model. It is built upon pytorch-geometric and provides usage with thenuScenesandRadarScenesdataset. Results Results of our RadarGNN model for object detection and semanti...