34 Commits

Author SHA1 Message Date
19e538bdff Implement API for predictions 2023-02-16 13:53:25 +01:00
8a9c3c1edc Change order of imports to avoid segfault 2023-02-16 13:52:01 +01:00
6f1dac0627 Change directory structure 2023-02-14 11:42:39 +01:00
55011144fb Remove autofocus sample 2023-02-14 11:22:42 +01:00
fcb21ff0b9 Move weights to canonical location 2023-02-14 11:22:09 +01:00
f9de6912d8 Set up folder structure for package 2023-02-14 11:21:40 +01:00
3959e449eb Remove run target 2023-02-13 18:13:49 +01:00
f652017466 Remove unused imports and linker flags 2023-02-13 18:13:02 +01:00
48d2620a43 Save image as current.jpg 2023-02-13 17:58:59 +01:00
85fe839b01 Rename focus_pipeline to video_pipeline 2023-02-13 17:58:28 +01:00
97bc3ee5b3 Remove autofocus code 2023-02-13 17:56:25 +01:00
4842949676 Remove unused image pipeline 2023-02-13 17:55:14 +01:00
4e9fcd5d3f Add C++ version for image capture 2023-02-13 17:54:28 +01:00
c0b9022bfc Adjust formatting 2023-02-13 17:54:13 +01:00
0f0e3c414a Ignore torchscript binaries 2023-02-08 10:11:25 +01:00
dedc96d867 Add export option for YOLO 2023-02-08 10:10:59 +01:00
dad740d92d Add autofocus to deployment 2023-02-08 10:08:49 +01:00
7847d546db Ignore datasets 2023-02-08 10:08:31 +01:00
6142f7af18 Organize code into modules 2023-02-08 10:08:12 +01:00
2e0d8b9364 Add tkinter import for commandline opencv 2023-02-08 10:05:09 +01:00
dfccc908c3 Ignore onnx models 2023-01-19 22:11:48 +01:00
93f3296eff Update paths of torch models 2023-01-19 22:10:34 +01:00
f9b2c25fe5 Move jetson deployment code 2023-01-19 21:36:58 +01:00
733e5f6fb6 Add type information to docstrings 2023-01-19 11:06:13 +01:00
548d7a1f9c Do not perform gradient calculation during inference 2023-01-19 11:02:32 +01:00
0ef8eb0cd4 Return the currently processed image 2023-01-19 10:58:44 +01:00
2f3d3da32c Add clarifying comments 2023-01-19 10:56:02 +01:00
618f80b564 Do not show image with bounding boxes 2023-01-19 10:55:39 +01:00
8e037586bc Remove comments 2023-01-19 10:53:14 +01:00
ad1bcc1dab Return dict of predictions 2023-01-19 10:52:31 +01:00
c735b01f8b Accept model weights as parameters 2023-01-19 10:51:08 +01:00
920c1fd852 Implement cropping and classification 2023-01-18 16:46:45 +01:00
e110aecc67 Delete detection runs 2023-01-18 16:46:15 +01:00
eb6546ddce Initial commit 2023-01-12 10:25:12 +01:00