Skip to content

CaesarG/TZB2023

Repository files navigation

RCS帧数据识别分类

骨干网使用Alexnet和VGG,针对rcs数据分别有2通道,4通道和ifft的识别模型,其中ifft识别效果最佳。

1.工程结构

dataRCS

1-10

存放1~10个类别的rcs数据,每个文件夹存放250帧的mat文件,每帧包含Ev,Eh两种极化接收方式,两种接收数据大小均为401x512,401对应8G到10GHz的采样点数, 512对应不同方位角俯仰角的接收位置。

annotations

存放训练集、验证集、测试集文件路径与类别的txt文件

dataset

annotation_gen.py

生成txt格式的标签文件

observer.py

观察转换后数据情况

dataset.py

直接abs接收数据的dataset,送入网络时是2x401x512

dataset_4.py

分开使用接收数据实部虚部的dataset,送入网络时是4x401x512

dataset_ifft.py

对频谱数据做ifft到时域后取abs的dataset,送入网络时是2x401x512

model

存放Alexnet,VGG网络及各种trick.

pretrain_weight

存放网络模型的与训练权重,加速网络训练速度,优化训练效果。Alexnet的预训练权值文件原链接丢失,目前方便获取的只有VGG权值。

test_result

存放test的预测结果

weight_of_model

存放训练结果,包括.pth文件和训练损失记录

eval.py

准确率测试

loss.py

损失函数,采用交叉熵损失函数

test.py

生成预测结果

train.py

训练文件

main.py

主文件,train,eval,test均在通过主文件完成

2.使用说明

注:batch_size根据电脑配置和模型大小调整,一般取2的次幂数

2通道ifft训练指令:

Alexnet:
python main.py --phase train --channel 2 --model Alexnet --ifft True --batch_size 16
VGG:
python main.py --phase train --channel 2 --model vgg --ifft True --batch_size 16

2通道ifft指标测试指令

Alexnet:
python main.py --phase eval --channel 2 --model Alexnet --ifft True --batch_size 16 --resume path(比如model_24.pth) --test_txt test.txt
VGG:
python main.py --phase eval --channel 2 --model vgg --ifft True --batch_size 16 --resume path(比如model_24.pth) --test_txt test.txt

2通道abs训练指令:

python main.py --phase train --channel 2 --model Alexnet --ifft False --batch_size 16

2通道指标测试指令:

python main.py --phase eval --channel 2 --model Alexnet --ifft False --batch_size 16 --resume path(比如model_24.pth) --test_txt test.txt

4通道ifft训练指令:

python main.py --phase train --channel 4 --model Alexnet --ifft True --batch_size 8

4通道ifft指标测试指令:

python main.py --phase eval --channel 4 --model Alexnet --ifft True --batch_size 8 --resume path(比如model_24.pth) --test_txt test.txt

4通道训练指令:

python main.py --phase train --channel 4 --model Alexnet --ifft False --batch_size 8

4通道指标测试指令:

python main.py --phase eval --channel 4 --model Alexnet --ifft False --batch_size 8 --resume path(比如model_24.pth) --test_txt test.txt

About

基于雷达RCS数据的空间物体智能识别技术研究

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages