mckay
|
d8b5b545e2
|
fix:修复loss参数
|
7 months ago |
mckay
|
bf6f473edd
|
refactor: 在networks文件里面移除loss相关的东西
|
7 months ago |
mckay
|
9f9b5816eb
|
feat: test脚本
|
7 months ago |
mckay
|
608c6b75bf
|
feat: 完善log记录,可以显示报错位置
|
7 months ago |
mckay
|
0eb2063e28
|
refactor: 模型相关函数拆分
|
7 months ago |
mckay
|
49e0f8e9bc
|
fix: git ignore add *.pth
|
7 months ago |
mckay
|
df186deae6
|
fix: can train
|
7 months ago |
mckay
|
1110dc220e
|
fix: 由于设置的采样点太小,取消Encoder1D的池化操作
|
7 months ago |
mckay
|
d839dcc091
|
todo: 配置修改
|
7 months ago |
mckay
|
2f269e2150
|
增加sdf dataloader随机采样
|
7 months ago |
mckay
|
66f214790c
|
fix :trian
|
7 months ago |
mckay
|
fe4f00a323
|
refactor: config配net
|
7 months ago |
mckay
|
fe9a5a68a7
|
TODO: 太大加载不了
|
7 months ago |
mckay
|
53c1f51826
|
fix :BRepFeatureEmbedder
|
7 months ago |
mckay
|
6361188e03
|
fix: sdf_xyz和sdf_value 分开加载
|
7 months ago |
mckay
|
6c452e1007
|
fix:使用size()而不是shape来获取维度
|
7 months ago |
王琛涵
|
3b494aa6e0
|
fix :train script
|
7 months ago |
mckay
|
271d237b66
|
fix: ignore wandb files
|
7 months ago |
mckay
|
055a3e6ee6
|
fix: config 路径使用绝对路径
|
7 months ago |
王琛涵
|
3de9a39ef5
|
fix : config refactor log & details separate
|
7 months ago |
王琛涵
|
e5e5034991
|
refactor & style: 用config配置train,同时增加保存checkpoint的信息
|
7 months ago |
王琛涵
|
e12c492d69
|
fix: net dim问题。num*3,变成[num,3],保留空间特征
|
7 months ago |
王琛涵
|
ee598e8340
|
refactor: 用config替代硬编码
|
7 months ago |
王琛涵
|
2aafbc87e3
|
refactor: process_brep,用config替代原来的100边采样
|
7 months ago |
王琛涵
|
b0b93b70ef
|
feat: 增加全局config
|
7 months ago |
mckay
|
4d4dbb7c7d
|
feat: train 脚本
|
7 months ago |
mckay
|
84fac4660c
|
test: 修改数据集加载测试
|
7 months ago |
mckay
|
31fcbecfea
|
fix: 修改数据加载process_brep_data,防止维度超过最大(用截取)
|
7 months ago |
mckay
|
bee51c9dc0
|
修复: 改进数据预处理中的数组填充逻辑
1. 修复了pad_zero函数中的负数维度问题
- 添加了对超长输入的截断处理
- 优化了掩码生成逻辑
- 保持多维数组的维度一致性
2. 主要改动:
- 处理数组长度 > max_len:进行截断
- 处理数组长度 < max_len:进行填充
- 处理数组长度 = max_len:直接返回
- 改进布尔掩码的生成方式
|
7 months ago |
mckay
|
dfaeca63ba
|
fix: 数据集可以加载
|
7 months ago |
mckay
|
295e707e2b
|
fix: 修改了check_data_format函数,正确检查对象数组类型
在parse_solid函数中确保所有数组都有正确的类型
对点云数据使用对象数组类型
对其他数据使用正确的数值类型(float32或int32)
|
7 months ago |
mckay
|
7ed86aee29
|
fix: corners维度变成[num_faces, max_edge, 2, 3]
|
7 months ago |
mckay
|
6f39b1f522
|
feat: log 不同level报错
|
7 months ago |
mckay
|
84f9ab91c5
|
增加log,修改corner
|
7 months ago |
mckay
|
8ac5b76a17
|
可以运行的process brep
|
7 months ago |
mckay
|
9a25fe7e6d
|
暂存
|
7 months ago |
mckay
|
1b8b92bf1d
|
data laod
|
7 months ago |
mckay
|
67a07d072d
|
fix: modules import
|
7 months ago |
王琛涵
|
9034e56e69
|
data
|
7 months ago |
王琛涵
|
9f169c7ae6
|
data fix test
|
7 months ago |
王琛涵
|
30a9e9aebf
|
加log
|
7 months ago |
王琛涵
|
043f991376
|
初步数据集dataset
|
7 months ago |
王琛涵
|
945db54c33
|
处理brep脚本可用
|
7 months ago |
王琛涵
|
7ba9eac542
|
查看pkl
|
7 months ago |
王琛涵
|
8f1b3e50d2
|
step2pkl
|
7 months ago |
王琛涵
|
ddc4808d02
|
encode.py可以运行
|
7 months ago |
mckay
|
c47dd74796
|
feat: 模型架构
|
7 months ago |
mckay
|
72fae292e1
|
数据集脚本
|
7 months ago |
mckay
|
115a4edfcb
|
init
|
7 months ago |
mckay
|
a60fd7cc6b
|
Initial commit
|
7 months ago |