v1.0

README.md

@@ -40,11 +40,11 @@ pip install -r requirement.txt
 <th valign="bottom">ViT-Large</th>
 <!-- TABLE BODY -->
 <tr><td align="left">Colour fundus image</td>
-<td align="center"><a href="https://dl.fbaipublicfiles.com/mae/pretrain/mae_pretrain_vit_large.pth">download</a></td>
+<td align="center"><a href="https://drive.google.com/file/d/1l62zbWUFTlp214SvK6eMwPQZAzcwoeBE/view?usp=sharing">download</a></td>
 </tr>
 <!-- TABLE BODY -->
 <tr><td align="left">OCT</td>
-<td align="center"><a href="https://dl.fbaipublicfiles.com/mae/pretrain/mae_pretrain_vit_large.pth">download</a></td>
+<td align="center"><a href="https://drive.google.com/file/d/1m6s7QYkjyjJDlpEuXm7Xp3PmjN-elfW2/view?usp=sharing">download</a></td>
 </tr>
 </tbody></table>
 

engine_finetune.py

@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import math

main_finetune.py

@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import argparse
@@ -203,11 +197,6 @@ def main(args):
         else:
             sampler_test = torch.utils.data.SequentialSampler(dataset_test)
             
-            
-    else:
-        sampler_train = torch.utils.data.RandomSampler(dataset_train)
-        sampler_val = torch.utils.data.SequentialSampler(dataset_val)
-        sampler_test = torch.utils.data.SequentialSampler(dataset_test)
 
     if global_rank == 0 and args.log_dir is not None and not args.eval:
         os.makedirs(args.log_dir, exist_ok=True)

models_vit.py

@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# timm: https://github.com/rwightman/pytorch-image-models/tree/master/timm
-# DeiT: https://github.com/facebookresearch/deit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 from functools import partial

util/datasets.py

@@ -1,15 +1,9 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import os
-import PIL
 from torchvision import datasets, transforms
 from timm.data import create_transform
 from timm.data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD

util/lr_decay.py

@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# ELECTRA https://github.com/google-research/electra
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import json

util/lr_sched.py

@@ -1,8 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
+# Partly revised by YZ @UCL&Moorfields
+# --------------------------------------------------------
 
 import math
 

util/misc.py

@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import builtins

util/pos_embed.py

@@ -1,10 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# Position embedding utils
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import numpy as np