{"id":22916841,"url":"https://github.com/chanmo/building-recognizer","last_synced_at":"2025-04-15T13:26:27.947Z","repository":{"id":236260373,"uuid":"718899486","full_name":"ChanMo/building-recognizer","owner":"ChanMo","description":"建筑物识别器是一种用于识别卫星图或航拍图中的建筑物的模型。该模型使用卷积神经网络 (CNN) 来提取图像的特征，然后使用分类器来识别图像中的建筑物。","archived":false,"fork":false,"pushed_at":"2023-12-02T08:45:21.000Z","size":1812,"stargazers_count":10,"open_issues_count":0,"forks_count":1,"subscribers_count":2,"default_branch":"main","last_synced_at":"2025-03-28T20:35:53.135Z","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":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/ChanMo.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"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}},"created_at":"2023-11-15T02:34:25.000Z","updated_at":"2025-03-24T11:24:13.000Z","dependencies_parsed_at":"2024-04-26T10:33:31.629Z","dependency_job_id":"ae9bc0ce-a016-41d7-8cdd-dcf4b855b117","html_url":"https://github.com/ChanMo/building-recognizer","commit_stats":null,"previous_names":["chanmo/building-recognizer"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ChanMo%2Fbuilding-recognizer","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ChanMo%2Fbuilding-recognizer/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ChanMo%2Fbuilding-recognizer/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/ChanMo%2Fbuilding-recognizer/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/ChanMo","download_url":"https://codeload.github.com/ChanMo/building-recognizer/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":249078627,"owners_count":21209221,"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":[],"created_at":"2024-12-14T06:14:48.521Z","updated_at":"2025-04-15T13:26:27.912Z","avatar_url":"https://github.com/ChanMo.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Building Recognizer\n\n建筑物识别器是一种用于识别卫星图或航拍图中的建筑物的模型。该模型使用卷积神经网络 (CNN) 来提取图像的特征，然后使用分类器来识别图像中的建筑物\n\n该模型的输入数据是卫星图或航拍图。输出数据是图像中建筑物的类别和边界框。\n\n![output.png](output.png)\n\n该模型可以用于各种应用，例如：\n\n* 城市规划：用于识别城市中的建筑物，以进行规划和管理。\n* 灾害管理：用于识别灾害发生后受损的建筑物，以进行救援和重建。\n* 环境监测：用于识别环境中的建筑物，以进行监测和保护。\n\n建筑物识别器是一项具有重要应用价值的技术。随着卫星图像和航拍图像技术的不断发展，建筑物识别器将在未来得到更加广泛的应用。\n\n\n## 模型选择\n\n模型使用的是Mask-RCNN\n```\nmodel = torchvision.models.detection.maskrcnn_resnet50_fpn()\n\n```\n\n## 数据集\n\n数据集使用的是CrowdAI的MappingChallenge中的数据集, 此数据集使用和COCO相同的格式, 可以直接通过CocoDetection加载\n```\ndataset = datasets.CocoDetection(\n    IMAGES_PATH,\n    ANNOTATIONS_PATH,\n    ...\n)\n```\n\n## 为模型增加新的特征值\n\n数据集中建筑物的Label是100\n```\nnum_classes = 101\nin_features = model.roi_heads.box_predictor.cls_score.in_features\nmodel.roi_heads.box_predictor = FastRCNNPredictor(in_features, num_classes)\n```\n\n## Reference:\n\n* https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html\n* https://github.com/crowdAI/crowdai-mapping-challenge-mask-rcnn\n* https://www.aicrowd.com/challenges/mapping-challenge/dataset_files\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fchanmo%2Fbuilding-recognizer","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fchanmo%2Fbuilding-recognizer","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fchanmo%2Fbuilding-recognizer/lists"}