ytongxie / cotr Goto Github PK

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

cd nnUNet
pip install -e .

cd CoTr_package
pip install -e .

@article{xie2021cotr,
  title={CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentation},
  author={Xie, Yutong and Zhang, Jianpeng and Shen, Chunhua and Xia, Yong},
  booktitle={MICCAI},
  year={2021}
}
  

Contributors

Stargazers

Forkers

jianpengz apple3c ghali007 ericyu97 benjamesbabala min-sheng abcxubu hired-ld dhtc pgu-nd magnety borhon yangyang117 kissandrun mayinjin rrrfrr chenjiashuo123 congvmit locchio akonepoint ltumorsegmentationtest ll-c8 eliseyang chengjianhong 111hxn rockeycoss myyyr qianjinfighter renatoseb huasheng-hou liming680 w1a2ng chadkowski cv-seg estellexia steve-zeyu-zhang peace-2018 woodyyyyyyyy yux11 git-tengsun tarekegn82 miraclemountain ppandahead lishang666 kamil-kaczor

        ref_d, ref_y, ref_x = torch.meshgrid(torch.linspace(0.5, D_ - 0.5, D_, dtype=torch.float32, device=device),
                                             torch.linspace(0.5, H_ - 0.5, H_, dtype=torch.float32, device=device),
                                             torch.linspace(0.5, W_ - 0.5, W_, dtype=torch.float32, device=device))

        ref_d = ref_d.reshape(-1)[None] / (valid_ratios[:, None, lvl, 0] * D_)
        ref_y = ref_y.reshape(-1)[None] / (valid_ratios[:, None, lvl, 2] * H_)
        ref_x = ref_x.reshape(-1)[None] / (valid_ratios[:, None, lvl, 1] * W_)

        ref = torch.stack((ref_d, ref_x, ref_y), -1)   # D W H
        reference_points_list.append(ref)
    reference_points = torch.cat(reference_points_list, 1)
    reference_points = reference_points[:, :, None] * valid_ratios[:, None]
    return reference_points