diff --git a/README.md b/README.md index f83f5ee..43e2bf5 100644 --- a/README.md +++ b/README.md @@ -245,7 +245,7 @@ These tag descriptions files come from the [RAM++](https://arxiv.org/abs/2310.15 1 RAM++ (14M) - Swin-Base + Swin-Large COCO, VG, SBU, CC3M, CC3M-val, CC12M Provide strong image tagging ability for any category. Download link diff --git a/inference_ram.py b/inference_ram.py index 6fae070..4cc80f7 100644 --- a/inference_ram.py +++ b/inference_ram.py @@ -28,7 +28,7 @@ default=384, type=int, metavar='N', - help='input image size (default: 448)') + help='input image size (default: 384)') if __name__ == "__main__": diff --git a/inference_tag2text.py b/inference_tag2text.py index 8962446..43ea302 100644 --- a/inference_tag2text.py +++ b/inference_tag2text.py @@ -28,7 +28,7 @@ default=384, type=int, metavar='N', - help='input image size (default: 448)') + help='input image size (default: 384)') parser.add_argument('--thre', default=0.68, type=float,