Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
iDTer
122e497b8d
|
1 year ago | |
|---|---|---|
| .. | ||
| __pycache__ | 1 year ago | |
| contents | 1 year ago | |
| textures | 1 year ago | |
| all_faces.txt | 1 year ago | |
| audi_et_te.mtl | 1 year ago | |
| audi_et_te.obj | 1 year ago | |
| data_loader.py | 1 year ago | |
| grad_cam.py | 1 year ago | |
| nmr_test.py | 1 year ago | |
| readme.md | 1 year ago | |
| train.py | 1 year ago | |
readme.md
Physical Attack
运行前依赖
- 数据集
- 3d obj 文件，和对应的 mtl纹理文件
- 需要被训练的face的id列表（保存成txt）
- 被训练的face对应的初始内容纹理content，以及内容纹理对应的边缘mask纹理 canny
训练
python train.py [参数]
- 训练参数
- epoch
- lr
- batchsize
- lamb 内容loss的系数lambda
- d1 对边缘区域的保护系数
- d2 对非边缘区域的保护系数
- t 平滑loss的系数
- 路径参数
- obj 3d obj文件路径
- faces 可训练faces的文件路径
- datapath 数据集路径
- content 内容纹理路径
- canny 边缘mask纹理路径
训练结果存储在logs/目录下相应的文件夹内，其中包括：
- 训练过程中的一些图片输出
- cam.jpg gradcam图
- cam_b.jpg
- cam_p.jpg
- mask.png
- test_total.jpg 模型输入图片展示
- texture2.png
- loss.txt 训练过程中的loss变化
- texture.npy 训练好的纹理文件
测试
python test.py --texture=path_to_texture
测试结果存储在本目录下的acc.txt文件中，为使用resnet152模型测试的正确率acc的结果
No Description
Wavefront Object Python other