search

dddzg / up-detr

[TPAMI 2022 & CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers
Apache License 2.0
476 stars 71 forks source link

backbone的build时会构建position embeding，需要用到nestedtensor的mask，请问输入tensor类型的patch不会报错吗 #27

Closed Flyooofly closed 2 years ago

Flyooofly commented 2 years ago

backbone的build时会构建position embeding，需要用到nestedtensor的mask，请问输入tensor类型的不会报错吗，我看代码里没有做相应的分支处理 1659264187(1)