Skip to content

Commit ca4bea5

Browse files
committed
conflict
1 parent 2b5b2ce commit ca4bea5

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

tensorlayer/app/computer_vision.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,11 +51,10 @@ def __call__(self, input_data):
5151
if self.model_name == 'yolo4-mscoco':
5252
batch_data = yolo4_input_processing(input_data)
5353
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)
5754
pred_bbox = yolo4_output_processing(feature_maps)
5855
output = result_to_json(input_data, pred_bbox)
56+
elif self.model_name == 'lcn':
57+
output = self.model(input_data)
5958
else:
6059
raise NotImplementedError
6160

0 commit comments

Comments
 (0)