ISTR icon indicating copy to clipboard operation
ISTR copied to clipboard

ISTR: End-to-End Instance Segmentation with Transformers (https://arxiv.org/abs/2105.00637)

Results 23 ISTR issues
Sort by recently updated
recently updated
newest added

@hujiecpp I am using a custom COCO dataset. Please is there some way to extract mAP metric for each individual class/label when evaluating? For example, I have 2 classes of...

RuntimeError: CUDA out of memory. Tried to allocate 58.00 MiB (GPU 1; 10.76 GiB total capacity; 9.75 GiB already allocated; 25.56 MiB free; 9.89 GiB reserved in total by PyTorch)...

dist_url='tcp://127.0.0.1:49152', eval_only=False, machine_rank=0, num_gpus=1, num_machines=1, opts=[], resume=False) [08/25 11:15:41 detectron2]: Contents of args.config_file=projects/ISTR/configs/ISTR-AE-R50-3x.yaml: _BASE_: "Base-ISTR.yaml" MODEL: WEIGHTS: "detectron2://ImageNetPretrained/torchvision/R-50.pkl" RESNETS: DEPTH: 50 STRIDE_IN_1X1: False ISTR: NUM_PROPOSALS: 300 NUM_CLASSES: 5 MASK_ENCODING_METHOD: "AE"...

return torch._C._nn.upsample_nearest2d(input, output_size, scale_factors) RuntimeError: CUDA out of memory. Tried to allocate 180.00 MiB (GPU 0; 10.92 GiB total capacity; 9.74 GiB already allocated; 57.56 MiB free; 9.90 GiB reserved...

after the training i got this result !! ![Screenshot from 2022-08-25 22-34-33](https://user-images.githubusercontent.com/112083938/186871668-c08f2ba2-d8f1-4a0b-a6e8-a551dcf26ea6.png)

where is the ISTR-R50-3x.yaml file ? can you share it please ?

您好,非常感谢您的工作,我在按照您的安装步骤尝试以后 遇到了以下问题: ![dd5888f4079d19125b9d965b1338d1a](https://github.com/hujiecpp/ISTR/assets/82853287/4967e7b5-65d3-412e-b7f6-2346727626b8) 我的安装步骤完全是按照下面的步骤的: ![f74e5c8377c361bddcf129dc3efca62](https://github.com/hujiecpp/ISTR/assets/82853287/e6292934-8ef3-4c83-9b7f-a9f03969012c) 可以麻烦您看看原因吗?万分感谢!

Hello, Thank you very much for sharing the code for the model, I'm trying to follow the installation guide. However, when I run the command `python setup.py build develop`, I...

detectron is difficult to install....