allow inputs with 3 channels (use_xyz)

2026-08-14 12:00:44 +08:00 · 2018-04-10 13:28:20 +08:00
parent 20af9060dd
commit 0a6b08b83f
6 changed files with 94 additions and 17 deletions
@@ -104,7 +104,7 @@ if __name__ == "__main__":
        shuffle=True
    )

-    model = Pointnet(num_classes=13)
+    model = Pointnet(num_classes=13, use_xyz=False)
    model.cuda()
    optimizer = optim.Adam(
        model.parameters(), lr=args.lr, weight_decay=args.weight_decay