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
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 ...
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...