easton-cau / SOTR

SOTR: Segmenting Objects with Transformers
MIT License
193 stars 32 forks source link

pytorch版本 #8

Closed roar-1128 closed 3 years ago

roar-1128 commented 3 years ago

只有pytorch1.6版本对应有detectron2==0.2.1. pytorch>1.7版本对应detectron2==0.5

但是要求pytorch>1.7,detectron2==0.2.1.

请问应该如何解决呢 期待作者的回复

easton-cau commented 3 years ago

Our environment:

torch 1.7.1 torchvision 0.8.2 detectron2 0.2.1

Build detectron2 from https://github.com/facebookresearch/detectron2/releases.

roar-1128 commented 3 years ago

torch1.7.1中没有0.2.1版本的detectron2

easton-cau commented 3 years ago

image

1.download detectron2-0.2.1.zip 2.unzip detectron2-0.2.1.zip 3.cd detectron2 && pip install -e .