Commit Graph

40 Commits

Author SHA1 Message Date
Nils Koch
ed80c3056f frontend/__init__.py: major refactor, still wip
now uses async to pipeline images for inference, triggered by signals, generates pointclouds from depth and displays them
2022-06-14 19:50:55 +02:00
Nils Koch
1eefa2847b frontend/kinect.py: script for live capture and triggering inference 2022-06-14 19:50:02 +02:00
Nils Koch
11959eef61 dataset.py: cleanup 2022-06-02 15:54:45 +02:00
Nils Koch
6ba2cd9e5d test_model.py: cleanup 2022-06-02 15:20:28 +02:00
Nils Koch
7e0305ed91 frontend/__init__.py: allow for data transfer minimization 2022-06-02 15:20:17 +02:00
Nils Koch
922bd3ab80 frontend/__init__.py: general improvements 2022-06-02 15:03:38 +02:00
Nils Koch
50581efa01 api_server.py: reformat, more customizability 2022-06-02 15:03:15 +02:00
Nils Koch
3bc0e7d575 frontend/__init__.py: restructure 2022-06-02 14:41:12 +02:00
Nils Koch
b614fcfd74 frontend/__init__.py: loop endlessly, show more info, allow model change 2022-06-02 12:13:29 +02:00
Nils Koch
bb15dcd0a1 api_server.py: report timing info, allow on the fly changing of model 2022-06-02 12:12:53 +02:00
Nils Koch
7219fdef7c test_model.py: cleanup 2022-06-02 12:12:26 +02:00
Nils Koch
1bdf2e7776 api_server.py: rework so it actually kinda works
results are weirdly poor, but idk
2022-06-01 16:55:39 +02:00
Nils Koch
46a6ae44af frontend/__init__.py: initial commit
works, but results are poor
2022-06-01 09:57:29 +02:00
Nils Koch
d1ee2bc4f2 api_server.py: fix model path 2022-05-31 11:57:12 +02:00
Nils Koch
ed71c16912 api_server.py: add missing device 2022-05-31 11:56:54 +02:00
Nils Koch
8f0c3a32b8 api_server.py: moved from frontend/main.py, made roughly functional 2022-05-31 11:53:12 +02:00
Nils Koch
3f48276aed frontend/main.py: initial commit
most likely borked, but needs testing
2022-05-30 16:41:08 +02:00
Nils Koch
9740e5d647 test_model.py: reformat 2022-05-30 16:27:13 +02:00
Nils Koch
17bf30fa2a adjust config 2022-05-30 16:23:47 +02:00
Nils Koch
70e4bf6fe1 add wandb, make compatible with ctd data 2022-05-30 16:13:06 +02:00
Ibai
b333d5515f Removed model 2022-04-12 10:13:13 +09:00
Ibai Gorordo
d91a867d5d
Merge pull request #2 from Qjizhi/main
enable training
2022-04-12 10:12:23 +09:00
wangxiaolong_S21009
a35086dc02 enable training 2022-04-11 11:23:00 -07:00
Ibai
982b5027b9 Fix flow_width 2022-04-08 21:52:13 +09:00
Ibai
d2c55bb097 Add ONNX conversion script 2022-04-08 20:46:00 +09:00
Ibai
2109614595 Merge branch 'main' of https://github.com/ibaiGorordo/CREStereo-Pytorch 2022-04-08 20:45:35 +09:00
Ibai
c432996dd5 Set flow_init to optional 2022-04-08 20:45:33 +09:00
Ibai Gorordo
fef3e4b3a0
Update README.md 2022-04-08 20:06:54 +09:00
Ibai
d199d62b8d Adjusted mesgh_grid for ONNX conversion 2022-04-08 18:10:20 +09:00
Ibai
f0a57bb444 Fix implementation issues 2022-04-08 17:48:29 +09:00
Ibai
66127ec2aa Update test_model.py 2022-04-08 12:21:56 +09:00
Ibai
7e43c65e01 Merge branch 'main' of https://github.com/ibaiGorordo/CREStereo-Pytorch 2022-04-08 12:00:39 +09:00
Ibai
3396d2fd06 Fix CUDA inference 2022-04-08 12:00:32 +09:00
Ibai Gorordo
7122ba4255
Update README.md 2022-04-08 11:36:53 +09:00
Ibai
494d12893b Update output.jpg 2022-04-08 11:30:01 +09:00
Ibai
c196d6caa3 Update output image 2022-04-08 11:27:29 +09:00
Ibai
1bdbb27240 #1 Added weight conversion to Pytorch 2022-04-08 11:17:32 +09:00
Ibai Gorordo
ecd3a40505
Update test_model.py 2022-04-08 01:17:45 +09:00
Ibai Gorordo
877bf354b9
Update README.md 2022-04-08 00:26:40 +09:00
Ibai
fc524edd76 Initial commit 2022-04-08 00:18:27 +09:00