From 2f5ce4d9d1c5bd9b8c7aa395dd0ad53ff9bbd7ee Mon Sep 17 00:00:00 2001 From: Mahdi Amrollahi Date: Mon, 28 Nov 2022 13:06:19 +0330 Subject: [PATCH] fix image sizes which were in tuple --- .gitignore | 14 ++++++++++++++ models/ar_model.py | 4 ++-- requirements.txt | 3 ++- 3 files changed, 18 insertions(+), 3 deletions(-) create mode 100644 .gitignore diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..cb1203e --- /dev/null +++ b/.gitignore @@ -0,0 +1,14 @@ + +*.pyc + +resnet_640x640/coco_det_finetune_resnet_640x640_checkpoint + +resnet_640x640/coco_det_finetune_resnet_640x640_ev_object_detection_validation_p0.4cocoeval.pkl + +resnet_640x640/coco_det_finetune_resnet_640x640_ev_object_detection_validation_p0.4_result.json + +resnet_640x640/coco_det_finetune_resnet_640x640_config.json + +resnet_640x640/coco_det_finetune_resnet_640x640_ckpt-71148.index + +resnet_640x640/coco_det_finetune_resnet_640x640_ckpt-71148.data-00000-of-00001 diff --git a/models/ar_model.py b/models/ar_model.py index 39b51f1..c01106b 100644 --- a/models/ar_model.py +++ b/models/ar_model.py @@ -42,14 +42,14 @@ def __init__(self, config: ml_collections.ConfigDict, **kwargs): mlp_ratio = config.dim_mlp // config.dim_att if config.resnet_variant == 'c1': self.encoder = VisionTransformer( - config.image_size[0], config.image_size[1], config.patch_size, + config.image_size, config.image_size, config.patch_size, config.num_encoder_layers, config.dim_att, mlp_ratio, config.num_heads, config.drop_path, config.drop_units, config.drop_att, config.pos_encoding, config.use_cls_token, name='vit') else: self.encoder = ResNetTransformer( - config.image_size[0], config.image_size[1], config.resnet_variant, + config.image_size, config.image_size, config.resnet_variant, config.resnet_depth, config.resnet_width_multiplier, config.resnet_sk_ratio, config.num_encoder_layers, config.dim_att, mlp_ratio, config.num_heads, config.drop_path, config.drop_units, diff --git a/requirements.txt b/requirements.txt index 6629173..6f7ad39 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,4 +5,5 @@ tensorflow-datasets tensorflow-addons tensorflow-text pycocotools -scikit-image \ No newline at end of file +scikit-image +numpy \ No newline at end of file