From ad280cbf1f323d18809d3979808c4da1ae2ca040 Mon Sep 17 00:00:00 2001 From: rmaphoh Date: Tue, 31 Jan 2023 00:47:18 +0000 Subject: [PATCH] v1.0 --- README.md | 4 ++-- engine_finetune.py | 8 +------- main_finetune.py | 13 +------------ models_vit.py | 8 +------- util/datasets.py | 8 +------- util/lr_decay.py | 8 +------- util/lr_sched.py | 5 ++--- util/misc.py | 8 +------- util/pos_embed.py | 6 +----- 9 files changed, 11 insertions(+), 57 deletions(-) diff --git a/README.md b/README.md index 0d560a5..b58a51c 100644 --- a/README.md +++ b/README.md @@ -40,11 +40,11 @@ pip install -r requirement.txt ViT-Large Colour fundus image -download +download OCT -download +download diff --git a/engine_finetune.py b/engine_finetune.py index f501738..54e7558 100644 --- a/engine_finetune.py +++ b/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 diff --git a/main_finetune.py b/main_finetune.py index 6c174d4..42c5330 100644 --- a/main_finetune.py +++ b/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) diff --git a/models_vit.py b/models_vit.py index 75cc90a..d7334c3 100644 --- a/models_vit.py +++ b/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 diff --git a/util/datasets.py b/util/datasets.py index f552c83..0e58750 100644 --- a/util/datasets.py +++ b/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 diff --git a/util/lr_decay.py b/util/lr_decay.py index 7fa11f1..c3b4eb5 100644 --- a/util/lr_decay.py +++ b/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 diff --git a/util/lr_sched.py b/util/lr_sched.py index 4cb682b..7a3a107 100644 --- a/util/lr_sched.py +++ b/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 diff --git a/util/misc.py b/util/misc.py index 50123b1..a79d696 100644 --- a/util/misc.py +++ b/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 diff --git a/util/pos_embed.py b/util/pos_embed.py index 6acf8bd..11f7128 100644 --- a/util/pos_embed.py +++ b/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