{"id":19603899,"url":"https://github.com/divelab/aeanets","last_synced_at":"2026-02-09T12:31:54.586Z","repository":{"id":77868660,"uuid":"454539350","full_name":"divelab/AEANets","owner":"divelab","description":"AEANets","archived":false,"fork":false,"pushed_at":"2022-10-14T21:32:13.000Z","size":44670,"stargazers_count":4,"open_issues_count":0,"forks_count":0,"subscribers_count":0,"default_branch":"main","last_synced_at":"2025-04-27T19:45:50.052Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"gpl-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/divelab.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2022-02-01T20:25:14.000Z","updated_at":"2024-05-22T20:09:46.000Z","dependencies_parsed_at":"2023-02-28T02:16:57.583Z","dependency_job_id":null,"html_url":"https://github.com/divelab/AEANets","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/divelab/AEANets","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/divelab%2FAEANets","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/divelab%2FAEANets/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/divelab%2FAEANets/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/divelab%2FAEANets/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/divelab","download_url":"https://codeload.github.com/divelab/AEANets/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/divelab%2FAEANets/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":29265202,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-02-09T04:11:57.159Z","status":"ssl_error","status_checked_at":"2026-02-09T04:11:56.117Z","response_time":56,"last_error":"SSL_connect returned=1 errno=0 peeraddr=140.82.121.5:443 state=error: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"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":[],"created_at":"2024-11-11T09:33:33.147Z","updated_at":"2026-02-09T12:31:54.578Z","avatar_url":"https://github.com/divelab.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Augmented Equivariant Attention Networks for Microscopy Image Transformation\n\nThis is the official implementation of AEANet in the paper [Augmented Equivariant Attention Networks for Microscopy Image Transformation](https://ieeexplore.ieee.org/stamp/stamp.jsp?arnumber=9785968) accepted by TMI.\n\n\u003cp align=\"center\"\u003e\n  \u003cimg src=\"ba-lib.png\" alt=\"drawing\" width=70%/\u003e\n\u003c/p\u003e\n\n## System Requirements and Environmental setup\n\nWe use the same system and environment as the work GVTNets. You can follow the insturctions in [this repo](https://github.com/divelab/GVTNets/) to setup the enviroment.\n\n## Usage\n\nYou can follow the example in this [Jypyter notebook](https://github.com/divelab/AEANets/blob/main/model3_pooled_batchatt_lib_kshape.ipynb) for the training and prediction using AEANets to reproduce the results in our paper or on your own data. We have provided the training checkpoints to reproduce the results.\n\nTo train the model, simply uncomment the following line in Cell 3\n\n```\nmodel.train(sources, targets, [256,256], validation=None, steps=120000, batch_size=8,seed=1)\n```\n\nThe arguments for the `model.train()` methods include\n\n- source_lst: a numpy array of training low-quality images of shape [N, W, H, C].\n- target_lst: a numpy array of training high-quality images of shape [N, W, H, C], in same order to source images.\n- patch_size: the patch size used for training. Will randomly crop training images into patches.\n- validation: [Optional] A tuple of (source, target) pair as the validation set.\n\nTo evaluate the model, use the `evaluate_mean_wh` function. Specifically, `evaluate_mean_wh(None)` performs prediction on the entire given image and generally produces the best results. Please refer to Figure 9 in Appendix VII for more discussions.\n\n\n## Bibtex\n\nIf you use our code, please consider cite our paper\n\n```\n@article{xie2022augmented,\n  title={Augmented Equivariant Attention Networks for Microscopy Image Transformation},\n  author={Xie, Yaochen and Ding, Yu and Ji, Shuiwang},\n  journal={IEEE Transactions on Medical Imaging},\n  year={2022},\n  publisher={IEEE}\n}\n```\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fdivelab%2Faeanets","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fdivelab%2Faeanets","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fdivelab%2Faeanets/lists"}