Commit Graph

596 Commits (3cad97418b27d8c66f9748f6b8ff6a44b69d6274)
 

Author SHA1 Message Date
zuchen.wang 3cad97418b generic 2021-11-24 16:00:17 +08:00
zuchen.wang e1d069abc7 Refactor: make pcb and clas in one sub project 2021-11-22 17:06:34 +08:00
zuchen.wang 9c4c6018a3 Arc+CrossEntropy/Focal Loss 2021-11-22 15:26:42 +08:00
zuchen.wang ae180917b4 comment tensorboard output and output cls_accuracy in train loop 2021-11-21 15:20:07 +08:00
zuchen.wang ceb7618f68 update requirments.txt 2021-11-21 15:15:13 +08:00
zuchen.wang f1e29e2d8b change to 1115 data 2021-11-19 14:39:35 +08:00
zuchen.wang dae8be0608 change log interval from 200 to 100 2021-11-19 14:39:10 +08:00
zuchen.wang c628c5d9ff fix binary cross entropy loss shape 2021-11-19 11:03:30 +08:00
zuchen.wang 2cdc57f5d2 optimize import 2021-11-18 17:24:30 +08:00
zuchen.wang 270520c864 change MultiStepLr intervals 2021-11-17 17:57:55 +08:00
zuchen.wang c2f0991139 不打印网络结构 2021-11-17 17:57:55 +08:00
zuchen.wang cf8315c5b3 改为1102数据 2021-11-17 17:57:48 +08:00
zuchen.wang 339b5f2188 auto port in dist_url 2021-11-12 20:21:40 +08:00
zuchen.wang bc63002d05 update online-pcb.yaml 2021-11-12 19:39:40 +08:00
zuchen.wang 781b989407 过滤0张或者1张图像 2021-11-12 15:25:43 +08:00
zuchen.wang fa20064a89 change activation in pcb_head: GELU -> ReLU 2021-11-12 15:24:01 +08:00
zuchen.wang 88cc245847 change interpolation from BICUBIC to BILINEAR 2021-11-12 15:22:45 +08:00
zuchen.wang d543b80dcf fix: Interpolation Warning 2021-11-10 17:20:52 +08:00
zuchen.wang 3be6d2c439 fix: pair_score_evaluator.py 2021-11-10 17:01:59 +08:00
zuchen.wang 93edc0cf53 change evaluator 2021-11-10 17:01:27 +08:00
zuchen.wang f8ae876c8c fix: 删除dataset相关的日志中令人迷惑的csv format 2021-11-10 17:00:50 +08:00
zuchen.wang a080930464 修改PCB最后一个全连接层unit: from 2 to 1 2021-11-10 16:56:39 +08:00
zuchen.wang dfd7e5f61e 添加binary cross entropy loss和binary focal loss 2021-11-10 16:55:40 +08:00
zuchen.wang da8b623ce8 调整阈值 2021-11-10 14:02:00 +08:00
zuchen.wang ee009f0a0d fix evaluator 2021-11-04 16:49:31 +08:00
zuchen.wang 1d8451a23a fix contrastive loss 2021-11-04 16:24:18 +08:00
zuchen.wang cc0b9e9612 fix contrastive loss 2021-11-04 13:55:40 +08:00
zuchen.wang 3d0ac7dc6b rm useless config yaml 2021-11-04 10:53:53 +08:00
zuchen.wang 3bbc80c0e7 empty commit: fix username 2021-11-04 10:41:36 +08:00
Sherlock 1342f5d599 add interpreter project 2021-11-04 10:41:36 +08:00
zuchen.wang 01538bec64 fix logger and pretty dataset info 2021-11-04 10:11:16 +08:00
zuchen.wang 21c14f1494 重构数据集 2021-11-03 17:25:58 +08:00
zuchen.wang 65cfc515d9 实现和线上模型一模一样的PCBNet 2021-11-02 17:56:05 +08:00
zuchen.wang 5a075c1fe8 Merge branch 'develop' into pcb_online 2021-10-29 14:29:29 +08:00
zuchen.wang d03eae9daa Implement online pcb config 2021-10-29 14:13:45 +08:00
zuchen.wang c90792042a make common base config: input, dataloader, solver, test period and batch 2021-10-29 11:45:24 +08:00
zuchen.wang 99b124304f change input size to (0,0) 2021-10-29 11:36:55 +08:00
zuchen.wang 44a90ae025 实现和线上一模一样的pcb 2021-10-29 09:45:59 +08:00
zuchen.wang eabe896ee8 train pcb, 未达到期望 2021-10-28 22:39:54 +08:00
zuchen.wang ad31a2cd16 modify config 2021-10-27 16:15:58 +08:00
zuchen.wang 1db050b51f fix evaluator 2021-10-27 16:15:37 +08:00
zuchen.wang 3880b23ab4 refactor evaluator 2021-10-27 16:03:43 +08:00
zuchen.wang 6687df06e0 update default config 2021-10-27 15:24:25 +08:00
zuchen.wang d4e2ac32d8 add pcb 2021-10-27 15:09:57 +08:00
zuchen.wang 31d7a51b62 fix bug in online_dataset.py 2021-10-22 13:31:12 +08:00
zuchen.wang d6768e2118 empty commit: fix username 2021-10-22 13:23:09 +08:00
l1aoxingyu d73fc53f8a feat: let compile evaluation cython automatically 2021-10-22 13:08:33 +08:00
zuchen.wang ae842114c7 update dataset config in trainer.py 2021-10-21 10:50:38 +08:00
zuchen.wang 863cb288f7 Merge branch 'master' into wzc/shoe 2021-10-20 18:46:50 +08:00
zuchen.wang 63233fecb0 add Cython in requirments 2021-10-20 18:46:39 +08:00