We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2b5b2ce commit ca4bea5Copy full SHA for ca4bea5
1 file changed
tensorlayer/app/computer_vision.py
@@ -51,11 +51,10 @@ def __call__(self, input_data):
51
if self.model_name == 'yolo4-mscoco':
52
batch_data = yolo4_input_processing(input_data)
53
feature_maps = self.model(batch_data, is_train=False)
54
- output = yolo4_output_processing(feature_maps)
55
- elif self.model_name == 'lcn':
56
- output = self.model(input_data)
57
pred_bbox = yolo4_output_processing(feature_maps)
58
output = result_to_json(input_data, pred_bbox)
+ elif self.model_name == 'lcn':
+ output = self.model(input_data)
59
else:
60
raise NotImplementedError
61
0 commit comments