119 Commits

Author SHA1 Message Date
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