SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
https://praeclarumjj3.github.io/oneformer
MIT License
1.41k stars 128 forks source link

Poor performance for instance task #22

Closed assia855 closed 1 year ago

assia855 commented 1 year ago

Thanks for sharing this work. I test the model and I have a similar result for panoptic and semantic but I have a poor performance for instance task. output

I tried to test the model with the swin backbone but I have this error:

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

I would like to train the model on custom datasets for instance segmentation can you provide please a demo for training on custom datasets.

Thanks in advance

praeclarumjj3 commented 1 year ago

Hi @assia855, thanks for your interest in our work. Could you share the command that you executed for the instance segmentation result?

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

This error usually means that the checkpoint is not downloaded completely. Did you download the checkpoint file correctly? It should be around 900 MB in size for the Swin-L backbone.

I would like to train the model on custom datasets for instance segmentation can you provide please a demo for training on custom datasets.

We share guidelines to train OneFormer on custom datasets here. You may also refer to #17.

assia855 commented 1 year ago

Hi @praeclarumjj3 I implement the colab demo locally and I had this output provide above for instance task. My environment:

Build cuda_11.3.r11.3 torch: 1.10 ; cuda: 1.10.1 detectron2: 0.6

I still have this error when I select swin True and launch the cell to load the models. Which checkpoint I should download for this one please.

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

Thanks for sharing the guideline.

praeclarumjj3 commented 1 year ago

@assia855 are you sure you did not make any changes (setup, predictor loading) to the colab code? I can successfully run the demos for all tasks using our colab demo.

assia855 commented 1 year ago

Thanks @praeclarumjj3 I will retry.