{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/infrared-small-target-detection-with-scale","title":"Infrared Small Target Detection with Scale and Location Sensitivity","arxiv_id":"2403.19366","date":"2024-03-28","proceeding":"CVPR 2024 1","authors":["Qiankun Liu","Rui Liu","Bolun Zheng","Hongkui Wang","Ying Fu"],"abstract":"Recently, infrared small target detection (IRSTD) has been dominated by deep-learning-based methods. However, these methods mainly focus on the design of complex model structures to extract discriminative features, leaving the loss functions for IRSTD under-explored. For example, the widely used Intersection over Union (IoU) and Dice losses lack sensitivity to the scales and locations of targets, limiting the detection performance of detectors. In this paper, we focus on boosting detection performance with a more effective loss but a simpler model structure. Specifically, we first propose a novel Scale and Location Sensitive (SLS) loss to handle the limitations of existing losses: 1) for scale sensitivity, we compute a weight for the IoU loss based on target scales to help the detector distinguish targets with different scales: 2) for location sensitivity, we introduce a penalty term based on the center points of targets to help the detector localize targets more precisely. Then, we design a simple Multi-Scale Head to the plain U-Net (MSHNet). By applying SLS loss to each scale of the predictions, our MSHNet outperforms existing state-of-the-art methods by a large margin. In addition, the detection performance of existing detectors can be further improved when trained with our SLS loss, demonstrating the effectiveness and generalization of our SLS loss. The code is available at https://github.com/ying-fu/MSHNet.","url_abs":"https://arxiv.org/abs/2403.19366v1","url_pdf":"https://arxiv.org/pdf/2403.19366v1.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"infrared-small-target-detection-with-scale","repo_url":"https://github.com/ying-fu/mshnet","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[{"task_slug":"sensitivity","task_name":"Sensitivity"}],"methods":[{"method_slug":"concatenated-skip-connection","method_name":"Concatenated Skip Connection"},{"method_slug":"convolution","method_name":"Convolution"},{"method_slug":"focus","method_name":"Focus"},{"method_slug":"max-pooling","method_name":"Max Pooling"},{"method_slug":"relu","method_name":"ReLU"},{"method_slug":"u-net","method_name":"U-Net"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2403.19366","atlas_url":"https://app.syntology.ai/?focus=2403.19366","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2403.19366"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/ying-fu/mshnet","reach":{"status":"ok"}},{"provenance":"deterministic:regex_extraction","url":"https://github.com/ying-fu/MSHNet","reach":{"status":"ok"}}],"summary":{"ran":3,"ran_draft_wrong":1,"unverified":3},"by_repo_kind":{"official":{"samples":7,"ran":4,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":7,"samples":[{"code_sha256_prefix":"079cb4f82c53233e","entry":"Dice","repo":"ying-fu/mshnet","repo_kind":"official","path":"model/loss.py","file_url":"https://github.com/ying-fu/mshnet/blob/HEAD/model/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"079cb4f82c53233e"}},{"code_sha256_prefix":"890d01d1c2196cc2","entry":"LLoss","repo":"ying-fu/MSHNet","repo_kind":"official","path":"model/loss.py","file_url":"https://github.com/ying-fu/MSHNet/blob/HEAD/model/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"890d01d1c2196cc2"}},{"code_sha256_prefix":"7505ade10111da54","entry":"SLSIoULoss","repo":"ying-fu/MSHNet","repo_kind":"official","path":"model/loss.py","file_url":"https://github.com/ying-fu/MSHNet/blob/HEAD/model/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"7505ade10111da54"}},{"code_sha256_prefix":"21bbc1e09581187f","entry":"SoftIoULoss","repo":"ying-fu/mshnet","repo_kind":"official","path":"model/loss.py","file_url":"https://github.com/ying-fu/mshnet/blob/HEAD/model/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"21bbc1e09581187f"}},{"code_sha256_prefix":"33361edfc92a6db7","entry":"batch_intersection_union","repo":"ying-fu/mshnet","repo_kind":"official","path":"utils/metric.py","file_url":"https://github.com/ying-fu/mshnet/blob/HEAD/utils/metric.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"33361edfc92a6db7"}},{"code_sha256_prefix":"4dfe8308e4ca4fc5","entry":"batch_pix_accuracy","repo":"ying-fu/mshnet","repo_kind":"official","path":"utils/metric.py","file_url":"https://github.com/ying-fu/mshnet/blob/HEAD/utils/metric.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4dfe8308e4ca4fc5"}},{"code_sha256_prefix":"98adcbe4f9819fab","entry":"cal_tp_pos_fp_neg","repo":"ying-fu/mshnet","repo_kind":"official","path":"utils/metric.py","file_url":"https://github.com/ying-fu/mshnet/blob/HEAD/utils/metric.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"98adcbe4f9819fab"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}