Code Monkey home page Code Monkey logo

istdu-net's Introduction

ISTDU-Net

code of article ISTDU-Net:Infrared Small-Target Detection U-Net

Introduction

This code using Python3.8+Pytorch1.8 is the complete code of article ISTDU-Net:Infrared Small-Target Detection U-Net, suitable for Ubuntu system.

Demo

demo.py:Input picture or video path, frame by frame output algorithm results(the code used to save the results is masked by default, please make your own modifications as required) The only thing that needs to be changed is the path: Line10 : videoPath = './test/images' ##change to the path where you want to test the image

The trained weight path is(already configured in this code): ./save_pth/ISTDU_Net/best.pth

!!!Note:img, _ = processGray(img, scale=scale, inp_h=inpShape[1], inp_w=inpShape[0]) This function automatically fills images of any size to adapt to the input of the network. Therefore, the larger the input image, the larger the GPU resources will be consumed. The input image should preferably be 512 x 512 pixels, a graphics card of 2080Ti is enough.

##The link to download the paper:https://doi.org/10.1109/LGRS.2022.3141584

##Reference:Q. Hou, L. Zhang, F. Tan, Y. Xi, H. Zheng and N. Li, "ISTDU-Net: Infrared Small-Target Detection U-Net," in IEEE Geoscience and Remote Sensing Letters, vol. 19, pp. 1-5, 2022, Art no. 7506205, doi: 10.1109/LGRS.2022.3141584.

istdu-net's People

Contributors

zhanglw882 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.