Commit Graph

81 Commits

Author SHA1 Message Date
Thang Vu
4c1b2876ca
Merge pull request #16 from thangvubk/code_refactor
Code refactor
2022-04-16 01:22:47 +09:00
Thang Vu
29836f17a1 update license 2022-04-15 16:20:30 +00:00
Thang Vu
4dbfef048b remove submodule 2022-04-15 16:18:58 +00:00
Thang Vu
096c5e8749 support save results 2022-04-15 16:08:19 +00:00
Thang Vu
590b96b8aa add setup.py 2022-04-15 08:16:32 +00:00
Thang Vu
3b39cc3bf1 minor 2022-04-15 05:37:01 +00:00
Thang Vu
efe2ddb6b1 separate train val func 2022-04-15 05:31:38 +00:00
Thang Vu
f483165c2a fix mem leak 2022-04-14 10:49:32 +00:00
Thang Vu
80a663eec6 convert spconv1 to spconv2 checkpoint 2022-04-14 04:10:35 +00:00
Thang Vu
70c86093db log with distributed parallel 2022-04-13 14:51:38 +00:00
Thang Vu
f0802b75bb support multiprocess test 2022-04-13 14:49:34 +00:00
Thang Vu
11130f850d support distributed eval/test 2022-04-11 07:34:59 +00:00
Thang Vu
08eb82816b minor 2022-04-11 02:03:29 +00:00
Thang Vu
c68be64a35 fix load checkpoint ddp 2022-04-11 02:00:22 +00:00
Thang Vu
40572e2834 add train aug_prob remove val aug 2022-04-11 01:17:44 +00:00
Thang Vu
3332a478a7 add cuda cast wrapper 2022-04-11 01:17:13 +00:00
Thang Vu
3475ab88b9 support mix precision training 2022-04-10 02:57:11 +00:00
Thang Vu
c620cfc435 support distributed training 2022-04-10 01:36:22 +00:00
Thang Vu
4090322ae0 support spconv2 2022-04-09 04:42:43 +00:00
Thang Vu
a93a1c58bc make cluster voxelization randomness optional 2022-04-09 03:31:57 +00:00
Thang Vu
068d67b1d4 support evaluate during training 2022-04-09 03:17:01 +00:00
Thang Vu
f811c2f815 minor 2022-04-08 12:40:55 +00:00
Thang Vu
7d8d7ce10b format with pre-commit 2022-04-08 12:35:12 +00:00
Thang Vu
9a73b2feb1 add pre-commit 2022-04-08 11:58:58 +00:00
Thang Vu
2b2b431fda relink spconv 2022-04-08 11:14:44 +00:00
Thang Vu
192f4bdd89 reorganize code 2022-04-08 11:10:38 +00:00
Thang Vu
7a20b6d6ae clean up 2022-04-08 10:25:13 +00:00
Thang Vu
a9e41f3f79 update configs 2022-04-08 10:12:50 +00:00
Thang Vu
0a17ca2cd1 refactor bfs_cluster 2022-04-08 09:49:21 +00:00
Thang Vu
190415587a format cuda code with clang-format 2022-04-08 06:07:52 +00:00
Thang Vu
97f306635c refactor train instance 2022-04-08 06:07:27 +00:00
Thang Vu
b5cd952a9e remove torch.cuda.empty_cache() 2022-04-08 05:53:39 +00:00
Thang Vu
559e870615 Merge remote-tracking branch 'origin/main' into code_refactor 2022-04-08 03:25:13 +00:00
Thang Vu
f7a31c531e update variable names 2022-04-08 03:19:49 +00:00
Thang Vu
70f2b7454f refactor logging 2022-04-07 12:49:08 +00:00
Thang Vu
af2c982653 refactor train point wise net 2022-04-07 07:38:38 +00:00
Thang Vu
e5e813ab67 support load checkpoint with different tensor shape 2022-04-06 10:17:10 +00:00
Thang Vu
009cc7c1f0 refactor s3dis dataset 2022-04-06 10:14:29 +00:00
Thang Vu
91c58d1ce3 remove data prefetch 2022-04-06 04:18:47 +00:00
Thang Vu
733b5c48e2 refactor scannetv2 2022-04-05 12:10:20 +00:00
Thang Vu
f822a70dd5 update forward test s3dis 2022-04-05 08:58:56 +00:00
Thang Vu
b62be6b409 refactor forward test 2022-04-04 09:03:32 +00:00
Thang Vu
253ef54373
Update README.md 2022-04-04 18:02:28 +09:00
Thang Vu
79fa20f8bc
Update README.md 2022-03-31 12:54:38 +09:00
Thang Vu
e62bad7aa0
Merge pull request #11 from thangvubk/s3dis_visualization
add s3dis visualization
2022-03-30 17:10:54 +09:00
Thang Vu
cd13355776
Add visualization guide for both scannet and s3dis 2022-03-30 17:10:27 +09:00
Thang Vu
6eb4d8a77a
Create visualization.md 2022-03-30 17:08:24 +09:00
Thang Vu
e27b3b8e81 add s3dis visualization 2022-03-30 08:08:06 +00:00
Thang Vu
ab147d1358
Update README.md 2022-03-24 13:33:05 +09:00
Thang Vu
40615587f6 fix small bug due to clean up 2022-03-23 08:44:27 +00:00