drprojects / superpoint_transformer

Official PyTorch implementation of Superpoint Transformer introduced in [ICCV'23] "Efficient 3D Semantic Segmentation with Superpoint Transformer" and SuperCluster introduced in [3DV'24 Oral] "Scalable 3D Panoptic Segmentation As Superpoint Graph Clustering"
MIT License
545 stars 71 forks source link

Predict behavior (draft PR) #46

Closed mbendjilali closed 6 months ago

mbendjilali commented 9 months ago

What does this PR do?

This pull request is a first shot to the implementation of the predict function from pytorch_lightning. Therefore, a predict.yaml config and predict.pyscript (almost duplicates of eval.yaml and eval.py) were added, the src/datasets/base.py and src/datamodules/base.py scripts were modified to initialize the needed prediction transforms and datasets.

I tested it. Although the predict function doesn't crash, I don't really see what it's doing. I also suspect that it doesn't consider the prediction set as batched and therefore only runs on one batch, leaving the others unprocessed.

Before submitting

Did you have fun?

Make sure you had fun coding 🙃

drprojects commented 9 months ago

Hi @mbendjilali thanks for this draft. I am currently out of office, will look into this when I come back.

drprojects commented 6 months ago

Hi, I finally released the v2 of this project, which should address this issue :partying_face:

Closing this PR then.