Skip to content

Anquanzhi/CSAP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CSAP

  1. Requirements: TensorFlow 1.2 or above
  2. Running: Training: python CSAP --train Test: python CSAP --test
  3. Key parameters

CSAP.py

GO_ON_TRAINING: Whether to continue training from the last breakpoint, 1 is yes, 0 is no
LOAD_PREVIOUS_POS: Whether to load the coding information, if it is not the first run and the training data has not been changed, set it to 0, otherwise 1 for recoding
MASK_DATA_PATH: If the roi area information is included in the test image, you can import the file under the corresponding path and only retain the detection results in the area
var_list: The core parameters used for transfer learning in tf.train.AdamOptimizer. Single-step transfer learning is performed when it is set to self.detector.vars_d, and transitive transfer is performed when it is set to self.detector.vars_stage2

model_fpn.py

with tf.variable_scope(self.name) as scope: Fully use the training data under the current task for training
with tf.variable_scope('trains_1') as scope: Used as a single-step transfer learning and general analysis of the detection model features, only this part of the feature is trained
with tf.variable_scope('trains_2') as scope: The training part of transfer learning based on single-step transfer learning

Reference:Q. An, Z. Pan, H. You and Y. Hu, "Transitive Transfer Learning-Based Anchor Free Rotatable Detector for SAR Target Detection With Few Samples," in IEEE Access, vol. 9, pp. 24011-24025, 2021, doi: 10.1109/ACCESS.2021.3056663.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages