Merge pull request #1855 from flytocc/VAN_Classification

add paper citation
pull/1868/head
Wei Shengyu 2022-04-20 14:46:05 +08:00 committed by GitHub
commit 05770197c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions

View File

@ -1,4 +1,4 @@
# copyright (c) 2021 PaddlePaddle Authors. All Rights Reserve.
# copyright (c) 2022 PaddlePaddle Authors. All Rights Reserve.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@ -185,6 +185,11 @@ class OverlapPatchEmbed(nn.Layer):
class VAN(nn.Layer):
r""" VAN
A PaddlePaddle impl of : `Visual Attention Network` -
https://arxiv.org/pdf/2202.09741.pdf
"""
def __init__(self,
img_size=224,
in_chans=3,