윤영준
09-12
changed model input size from 640 640 to 480 480
@870ee36c008d7f0ef81ccf365fbd26d989622d0b
--- inference_endpoint.py
+++ inference_endpoint.py
... | ... | @@ -29,7 +29,7 @@ |
| 29 | 29 |
classes_txt_file = 'config_files/yolo_config.txt' |
| 30 | 30 |
image_path = 'yoloseg/img3.jpg' |
| 31 | 31 |
|
| 32 |
-model_input_shape = (640, 640) |
|
| 32 |
+model_input_shape = (480, 480) |
|
| 33 | 33 |
inference_engine = Inference( |
| 34 | 34 |
onnx_model_path=model_path, |
| 35 | 35 |
model_input_shape=model_input_shape, |
Add a comment
Delete comment
Once you delete this comment, you won't be able to recover it. Are you sure you want to delete this comment?