update

sd68 · Jun 17, 2020 · 7cee484 · 7cee484
1 parent 01b12d5
commit 7cee484
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/lanenet_model/lanenet_postprocess.py b/lanenet_model/lanenet_postprocess.py
@@ -350,7 +350,7 @@ def postprocess(self, binary_seg_result, instance_seg_result=None,
                 tmp_mask = np.zeros(shape=(720, 1280), dtype=np.uint8)
                 tmp_mask[tuple((np.int_(coords[:, 1] * 720 / 256), np.int_(coords[:, 0] * 1280 / 512)))] = 255
             else:
-                raise ValueError('Wrong data source now only support tusimple and beec_ccd')
+                raise ValueError('Wrong data source now only support tusimple')
             tmp_ipm_mask = cv2.remap(
                 tmp_mask,
                 self._remap_to_ipm_x,
@@ -390,11 +390,8 @@ def postprocess(self, binary_seg_result, instance_seg_result=None,
             if data_source == 'tusimple':
                 start_plot_y = 240
                 end_plot_y = 720
-            elif data_source == 'beec_ccd':
-                start_plot_y = 820
-                end_plot_y = 1350
             else:
-                raise ValueError('Wrong data source now only support tusimple and beec_ccd')
+                raise ValueError('Wrong data source now only support tusimple')
             step = int(math.floor((end_plot_y - start_plot_y) / 10))
             for plot_y in np.linspace(start_plot_y, end_plot_y, step):
                 diff = single_lane_pt_y - plot_y