search

SHI-Labs / Neighborhood-Attention-Transformer

Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
MIT License
1.04k stars 85 forks source link

how to do Visualization #40

Closed skyfallsss closed 2 years ago

skyfallsss commented 2 years ago

mmdet show the error NAT is not registered.I can't solve the problem.

alihassanijr commented 2 years ago

Hello and thanks for your interest. Have you followed the instructions on setting up the environment correctly?

skyfallsss commented 2 years ago

I can train and test the model successfully,but I want to show the result I tested,such as the painted pictures or the pictures with bounding boxes.

alihassanijr commented 2 years ago

We haven't included those scripts, but using them only requires you to move them to the detection/ directory, and add this line to the python script:

from nat import *
skyfallsss commented 2 years ago

I will take a try.Thank you for your help.The output file formed in pickle,how can I use it?

skyfallsss commented 2 years ago

In other word,how can I print the bouding box data by using the test script?

skyfallsss commented 2 years ago

Thank you for your help.I solved the problem by using from nat import *.