 RenLiqiang
|
9cf8d177b5
添加3个loihead的forward方法
|
před 5 měsíci |
 lstrlq
|
11be386265
Merge branch 'master' of https://dev.lstznkj.com/DevLibs/MultiVisionModels
|
před 5 měsíci |
|
lstrlq
|
d3e5377a16
train maxvitfpn on 4080
|
před 5 měsíci |
|
RenLiqiang
|
298422fcbf
添加swin_trans_former
|
před 5 měsíci |
|
RenLiqiang
|
b8ae0774f3
修改分离line_head 和point_head
|
před 5 měsíci |
|
lstrlq
|
ab087b6767
debug train line on 4080
|
před 5 měsíci |
|
RenLiqiang
|
e974ae5475
debug train line
|
před 5 měsíci |
|
RenLiqiang
|
f8de422289
重构代码，分离损失计算相关函数
|
před 5 měsíci |
|
lstrlq
|
05e2ac222b
add predictor(transpose conv to upsample),debug maxvitfpn output point
|
před 5 měsíci |
|
lstrlq
|
8ac0f02074
debug train line_detect_maxvitfpn
|
před 5 měsíci |
|
lstrlq
|
7972128d83
加入maxvitfpn backbone
|
před 5 měsíci |
|
lstrlq
|
832e737f65
修复整合point和line导致报错的bug，分离point、line推理相关功能
|
před 5 měsíci |
|
lstrlq
|
e5e1c3d916
single point 和line 进行初步整合
|
před 5 měsíci |
|
lstrlq
|
cf6ea4a3ad
debug single point on 4080
|
před 5 měsíci |
|
RenLiqiang
|
b5faea6834
调整inference输出的points结构，可直接调用drawkeypoint
|
před 5 měsíci |
|
RenLiqiang
|
aab8cf148a
调试point inference
|
před 5 měsíci |
|
RenLiqiang
|
fcf18376b5
修复single point 映射ROI失败bug
|
před 5 měsíci |
|
RenLiqiang
|
eb2fc524a4
调试single point,解决部分报错，可进行backward
|
před 5 měsíci |
|
RenLiqiang
|
e212a1fd73
调试single point
|
před 5 měsíci |
|
RenLiqiang
|
6cfebd9e97
添加动态数据增强功能
|
před 5 měsíci |
|
lstrlq
|
77a2228b4d
原分辨率特征图提取线段在4080取得初步效果
|
před 5 měsíci |
|
RenLiqiang
|
5338e7af68
添加线段角度、长度等损失,修复roi映射问题
|
před 5 měsíci |
|
RenLiqiang
|
af4cac8846
修复标注点映射问题
|
před 5 měsíci |
|
lstrlq
|
0f2cd3a38f
修复batchsize大于1时 报错bug
|
před 5 měsíci |
|
RenLiqiang
|
91ca516394
删除lines_roi_pool
|
před 5 měsíci |
|
RenLiqiang
|
6806148273
linedataset 添加数据增强
|
před 5 měsíci |
|
RenLiqiang
|
51ec3b67b1
line_detect添加直接读取xanylabel标注数据文件功能
|
před 5 měsíci |
 xue50
|
c59317204e
完善dataset，从train.yaml传入参数，rgb或单通道tiff
|
před 5 měsíci |
|
RenLiqiang
|
866fc733c3
修改lines_iou_loss为并行计算
|
před 5 měsíci |
|
lstrlq
|
0f19a393c5
修改line_iou_loss支持并行计算
|
před 5 měsíci |
