【文件属性】：
文件名称：ShootLessSketchMore:DDM专案
文件大小：29.89MB
文件格式：ZIP
更新时间：2021-04-10 23:03:04
Python
射击少素描更多
数据和文档挖掘项目。
创建允许执行少量学习（特别是使用可变N向的一键式学习）的暹罗网络。
如何运行代码
python3 train.py --dataset tuberlin
python3 train.py --dataset miniquickdraw
python3 evaluation.py --dataset miniquickdraw
python3 evaluation.py --dataset tuberlin
火车
训练一个暹罗网络，该网络从数据加载器获取两个图像（如果类别不同，则标签为0），如果类别相同，则标签为1。
学习速度为10 ^ -3的Adam优化器
Logits的二进制交叉熵
测试集80％，验证集20％
在执行结束时保存经过训练的模型，并显示带有损失趋势的图形
评估
加载先前训练的模型
数据加载器提供一个图像和一组N-Way图像，其中一个
【文件预览】：
ShootLessSketchMore-main
----DataUtils()
--------class_names.txt(5KB)
--------test.txt(176B)
--------load_data.py(2KB)
--------prepare_data.py(517B)
--------__pycache__()
--------train.txt(544B)
--------val.txt(174B)
--------MiniQuickDraw()
--------generate_data.py(4KB)
----train_val_splitter.py(2KB)
----TUBerlin()
--------val()
--------train()
--------test()
----miniquickdraw_model.pt(17.08MB)
----models.py(3KB)
----README.md(2KB)
----config.py(104B)
----dataset.py(1KB)
----main.py(1KB)
----evaluation.py(2KB)
----dataset_n_way.py(4KB)
----train.py(3KB)