mirrors
/
SegFormer
mirror of https://github.com/NVlabs/SegFormer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update

pull/1/head
xieenze 2021-06-13 11:42:27 +08:00
parent f6af29ff18
commit 7d90b186be
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mmseg/models/backbones/mix_transformer.py
View File

@ -1,3 +1,8 @@
# ---------------------------------------------------------------
# Copyright (c) 2021, NVIDIA Corporation. All rights reserved.
#
# This work is licensed under the NVIDIA Source Code License
# ---------------------------------------------------------------
import torch
import torch.nn as nn
import torch.nn.functional as F

5
mmseg/models/decode_heads/segformer_head.py
View File

@ -1,3 +1,8 @@
# ---------------------------------------------------------------
# Copyright (c) 2021, NVIDIA Corporation. All rights reserved.
#
# This work is licensed under the NVIDIA Source Code License
# ---------------------------------------------------------------
import numpy as np
import torch.nn as nn
import torch