{"id":13738130,"url":"https://github.com/adobe/antialiased-cnns","last_synced_at":"2025-05-14T19:04:53.349Z","repository":{"id":38325647,"uuid":"186698598","full_name":"adobe/antialiased-cnns","owner":"adobe","description":"pip install antialiased-cnns to improve stability and accuracy","archived":false,"fork":false,"pushed_at":"2024-04-08T12:49:27.000Z","size":77815,"stargazers_count":1671,"open_issues_count":17,"forks_count":204,"subscribers_count":36,"default_branch":"master","last_synced_at":"2025-04-12T02:59:26.154Z","etag":null,"topics":["antialiasing","artificial-intelligence","cnns","computer-vision","convolutional-neural-networks","icml","icml-2019","shift-equivariant","shift-invariant"],"latest_commit_sha":null,"homepage":"https://richzhang.github.io/antialiased-cnns/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"other","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/adobe.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":".github/CONTRIBUTING.md","funding":null,"license":"LICENSE.md","code_of_conduct":"CODE_OF_CONDUCT.md","threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null}},"created_at":"2019-05-14T20:51:25.000Z","updated_at":"2025-04-11T06:57:15.000Z","dependencies_parsed_at":"2024-04-11T21:48:48.266Z","dependency_job_id":"aea57049-a933-40af-89ff-7945dbc0c21e","html_url":"https://github.com/adobe/antialiased-cnns","commit_stats":{"total_commits":228,"total_committers":9,"mean_commits":"25.333333333333332","dds":0.4035087719298246,"last_synced_commit":"b27a34a26f3ab039113d44d83c54d0428598ac9c"},"previous_names":[],"tags_count":4,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/adobe%2Fantialiased-cnns","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/adobe%2Fantialiased-cnns/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/adobe%2Fantialiased-cnns/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/adobe%2Fantialiased-cnns/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/adobe","download_url":"https://codeload.github.com/adobe/antialiased-cnns/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":253805848,"owners_count":21967053,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["antialiasing","artificial-intelligence","cnns","computer-vision","convolutional-neural-networks","icml","icml-2019","shift-equivariant","shift-invariant"],"created_at":"2024-08-03T03:02:12.028Z","updated_at":"2025-05-14T19:04:53.306Z","avatar_url":"https://github.com/adobe.png","language":"Python","funding_links":[],"categories":["Python","Pytorch实用程序"],"sub_categories":[],"readme":"# \u003cb\u003eAntialiased CNNs\u003c/b\u003e [[Project Page]](http://richzhang.github.io/antialiased-cnns/) [[Paper]](https://arxiv.org/abs/1904.11486) [[Talk]](https://www.youtube.com/watch?v=HjewNBZz00w)\n\n\u003cimg src='https://richzhang.github.io/antialiased-cnns/resources/gifs2/video_00810.gif' align=\"right\" width=300\u003e\n\n**Making Convolutional Networks Shift-Invariant Again** \u003cbr\u003e\n[Richard Zhang](https://richzhang.github.io/). In [ICML, 2019](https://arxiv.org/abs/1904.11486).\n\n### Quick \u0026 easy start\n\nRun `pip install antialiased-cnns`\n\n```python\nimport antialiased_cnns\nmodel = antialiased_cnns.resnet50(pretrained=True) \n```\n\u003c!-- model.load_state_dict(torch.load('resnet50_lpf4-994b528f.pth.tar')['state_dict']) # load weights; download it beforehand from https://www.dropbox.com/s/zqsudi0oz5ym8w8/resnet50_lpf4-994b528f.pth.tar?dl=0 --\u003e\n\n\u003c!-- Now you are antialiased! --\u003e\n\nIf you have a model already and want to antialias and continue training, copy your old weights over:\n\n``` python\nimport torchvision.models as models\nold_model = models.resnet50(pretrained=True) # old (aliased) model\nantialiased_cnns.copy_params_buffers(old_model, model) # copy the weights over\n```\n\nIf you want to modify your own model, use the BlurPool layer. More information about our provided models and how to use BlurPool is below.\n\n```python\nC = 10 # example feature channel size\nblurpool = antialiased_cnns.BlurPool(C, stride=2) # BlurPool layer; use to downsample a feature map\nex_tens = torch.Tensor(1,C,128,128)\nprint(blurpool(ex_tens).shape) # 1xCx64x64 tensor\n```\n\n**Updates**\n* **(Oct 2020) Finetune** I initialize the antialiased model with weights from baseline model, and finetune. Before, I was training from scratch. The results are better.\n* **(Oct 2020) Additional models** We now have 23 total model variants. I added variants of vgg, densenet, resnext, wide resnet varieties! The same conclusions hold.\n* **(Sept 2020) Pip install** You can also now `pip install antialiased-cnns` and load models with the `pretrained=True` flag.\n* **(Sept 2020) Kernel 4** I have added kernel size 4 experiments. When downsampling an even sized feature map (e.g., a 128x128--\u003e64x64), this is actually the correct size to use to keep the indices from drifting.\n\n### Table of contents\n\n1. [More information about antialiased models](#1-more-information-loading-an-antialiased-model)\u003cbr\u003e\n2. [Instructions for antialiasing your own model](#2-more-information-how-to-antialias-your-own-architecture), using the [`BlurPool`](antialiased_cnns/__init__.py) layer\u003cbr\u003e\n3. [ImageNet training and evaluation code](README_IMAGENET.md). Achieving better consistency, while maintaining or improving accuracy, is an open problem. Help improve the results!\n\n## (0) Preliminaries\n\nPip install this package\n\n- `pip install antialiased-cnns`\n\nOr clone this repository and install requirements (notably, PyTorch)\n\n```bash\n\nhttps://github.com/adobe/antialiased-cnns.git\ncd antialiased-cnns\npip install -r requirements.txt\n```\n\n\n## (1) Loading an antialiased model\n\nThe following loads a pretrained antialiased model, perhaps as a backbone for your application.\n\n```python\nimport antialiased_cnns\nmodel = antialiased_cnns.resnet50(pretrained=True, filter_size=4)\n```\n\nWe also provide weights for antialiased `AlexNet`, `VGG16(bn)`, `Resnet18,34,50,101`, `Densenet121`, and `MobileNetv2` (see [example_usage.py](example_usage.py)).\n\n## (2) How to antialias your own architecture\n\nThe `antialiased_cnns` module contains the `BlurPool` [class](antialiased_cnns/downsample.py), which does blur+subsampling. Run `pip install antialiased-cnns` or copy the `antialiased_cnns` subdirectory.\n\n**Methodology** The methodology is simple -- first evaluate with stride 1, and then use our `BlurPool` layer to do antialiased downsampling. Make the following architectural changes.\n\n```python\nimport antialiased_cnns\n\n# MaxPool --\u003e MaxBlurPool\nbaseline = nn.MaxPool2d(kernel_size=2, stride=2)\nantialiased = [nn.MaxPool2d(kernel_size=2, stride=1), \n    antialiased_cnns.BlurPool(C, stride=2)]\n    \n# Conv --\u003e ConvBlurPool\nbaseline = [nn.Conv2d(Cin, C, kernel_size=3, stride=2, padding=1), \n    nn.ReLU(inplace=True)]\nantialiased = [nn.Conv2d(Cin, C, kernel_size=3, stride=1, padding=1),\n    nn.ReLU(inplace=True),\n    antialiased_cnns.BlurPool(C, stride=2)]\n\n# AvgPool --\u003e BlurPool\nbaseline = nn.AvgPool2d(kernel_size=2, stride=2)\nantialiased = antialiased_cnns.BlurPool(C, stride=2)\n```\n\nWe assume incoming tensor has `C` channels. Computing a layer at stride 1 instead of stride 2 adds memory and run-time. As such, we typically skip antialiasing at the highest-resolution (early in the network), to prevent large increases.\n\n**Add antialiasing and then continue training** If you already trained a model, and then add antialiasing, you can fine-tune from that old model:\n\n``` python\nantialiased_cnns.copy_params_buffers(old_model, antialiased_model)\n```\n\nIf this doesn't work, you can just copy the parameters (and not buffers). Adding antialiasing doesn't add any parameters, so the parameter lists are identical. (It does add buffers, so some heuristic is used to match the buffers, which may throw an error.)\n\n``` python\nantialiased_cnns.copy_params(old_model, antialiased_model)\n```\n\n\u003cimg src='https://richzhang.github.io/antialiased-cnns/resources/antialias_mod.jpg' width=800\u003e\u003cbr\u003e\n\n## (3) ImageNet Evaluation, Results, and Training code\n\nWe observe improvements in both **accuracy** (how often the image is classified correctly) and **consistency** (how often two shifts of the same image are classified the same).\n\n\u003cimg src='plots/plots2_acc.png' align=\"left\" width=750\u003e\n\n\u003cimg src='plots/plots2_con.png' align=\"left\" width=750\u003e\n\n| ACCURACY | Baseline | Antialiased | Delta |\n| :------: | :------: | :-------: | :-------: |\n| alexnet | 56.55 | 56.94 | +0.39 |\n| vgg11 | 69.02 | 70.51 | +1.49 |\n| vgg13 | 69.93 | 71.52 | +1.59 |\n| vgg16 | 71.59 | 72.96 | +1.37 |\n| vgg19 | 72.38 | 73.54 | +1.16 |\n| vgg11_bn | 70.38 | 72.63 | +2.25 |\n| vgg13_bn | 71.55 | 73.61 | +2.06 |\n| vgg16_bn | 73.36 | 75.13 | +1.77 |\n| vgg19_bn | 74.24 | 75.68 | +1.44 |\n| resnet18 | 69.74 | 71.67 | +1.93 |\n| resnet34 | 73.30 | 74.60 | +1.30 |\n| resnet50 | 76.16 | 77.41 | +1.25 |\n| resnet101 | 77.37 | 78.38 | +1.01 |\n| resnet152 | 78.31 | 79.07 | +0.76 |\n| resnext50_32x4d | 77.62 | 77.93 | +0.31 |\n| resnext101_32x8d | 79.31 | 79.33 | +0.02 |\n| wide_resnet50_2 | 78.47 | 78.70 | +0.23 |\n| wide_resnet101_2 | 78.85 | 78.99 | +0.14 |\n| densenet121 | 74.43 | 75.79 | +1.36 |\n| densenet169 | 75.60 | 76.73 | +1.13 |\n| densenet201 | 76.90 | 77.31 | +0.41 |\n| densenet161 | 77.14 | 77.88 | +0.74 |\n| mobilenet_v2 | 71.88 | 72.72 | +0.84 |\n\n| CONSISTENCY | Baseline | Antialiased | Delta |\n| :------: | :------: | :-------: | :-------: |\n| alexnet | 78.18 | 83.31 | +5.13 |\n| vgg11 | 86.58 | 90.09 | +3.51 |\n| vgg13 | 86.92 | 90.31 | +3.39 |\n| vgg16 | 88.52 | 90.91 | +2.39 |\n| vgg19 | 89.17 | 91.08 | +1.91 |\n| vgg11_bn | 87.16 | 90.67 | +3.51 |\n| vgg13_bn | 88.03 | 91.09 | +3.06 |\n| vgg16_bn | 89.24 | 91.58 | +2.34 |\n| vgg19_bn | 89.59 | 91.60 | +2.01 |\n| resnet18 | 85.11 | 88.36 | +3.25 |\n| resnet34 | 87.56 | 89.77 | +2.21 |\n| resnet50 | 89.20 | 91.32 | +2.12 |\n| resnet101 | 89.81 | 91.97 | +2.16 |\n| resnet152 | 90.92 | 92.42 | +1.50 |\n| resnext50_32x4d | 90.17 | 91.48 | +1.31 |\n| resnext101_32x8d | 91.33 | 92.67 | +1.34 |\n| wide_resnet50_2 | 90.77 | 92.46 | +1.69 |\n| wide_resnet101_2 | 90.93 | 92.10 | +1.17 |\n| densenet121 | 88.81 | 90.35 | +1.54 |\n| densenet169 | 89.68 | 90.61 | +0.93 |\n| densenet201 | 90.36 | 91.32 | +0.96 |\n| densenet161 | 90.82 | 91.66 | +0.84 |\n| mobilenet_v2 | 86.50 | 87.73 | +1.23 |\n\nTo reduce clutter, extended results (different filter sizes) are [here](README_IMAGENET.md). Help improve the results!\n\n## Licenses\n\n\u003ca rel=\"license\" href=\"http://creativecommons.org/licenses/by-nc-sa/4.0/\"\u003e\u003cimg alt=\"Creative Commons License\" style=\"border-width:0\" src=\"https://i.creativecommons.org/l/by-nc-sa/4.0/80x15.png\" /\u003e\u003c/a\u003e\u003cbr /\u003eThis work is licensed under a \u003ca rel=\"license\" href=\"http://creativecommons.org/licenses/by-nc-sa/4.0/\"\u003eCreative Commons Attribution-NonCommercial-ShareAlike 4.0 International License\u003c/a\u003e.\n\nAll material is made available under [Creative Commons BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode) license by Adobe Inc. You can **use, redistribute, and adapt** the material for **non-commercial purposes**, as long as you give appropriate credit by **citing our paper** and **indicating any changes** that you've made.\n\nThe repository builds off the PyTorch [examples repository](https://github.com/pytorch/examples) and torchvision [models repository](https://github.com/pytorch/vision/tree/master/torchvision/models). These are [BSD-style licensed](https://github.com/pytorch/examples/blob/master/LICENSE).\n\n## Citation, contact\n\nIf you find this useful for your research, please consider citing this [bibtex](https://richzhang.github.io/index_files/bibtex_icml2019.txt). Please contact Richard Zhang \\\u003crizhang at adobe dot com\\\u003e with any comments or feedback.\n\n\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fadobe%2Fantialiased-cnns","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fadobe%2Fantialiased-cnns","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fadobe%2Fantialiased-cnns/lists"}