oValach commited on
Commit
149b46d
1 Parent(s): 99329ae

Update TheDistanceAssessor.py

Browse files
Files changed (1) hide show
  1. TheDistanceAssessor.py +1 -1
TheDistanceAssessor.py CHANGED
@@ -909,7 +909,7 @@ def run(input_image, model_seg, model_det, image_size, target_distances, num_ys
909
 
910
  output_image = get_result(classification, id_map, model.names, borders, image, regions)
911
  cropped_image = output_image[22:output_image.shape[0] - 40, 74:output_image.shape[1] - 33]
912
- return output_image
913
 
914
  if __name__ == "__main__":
915
 
 
909
 
910
  output_image = get_result(classification, id_map, model.names, borders, image, regions)
911
  cropped_image = output_image[22:output_image.shape[0] - 40, 74:output_image.shape[1] - 33]
912
+ return cropped_image
913
 
914
  if __name__ == "__main__":
915