ContourNet test IC15 数据集生成可视化结果

最新推荐文章于 2024-05-31 21:08:00 发布

Enki08

最新推荐文章于 2024-05-31 21:08:00 发布

阅读量312

点赞数

分类专栏：目标检测文章标签：神经网络 pytorch 深度学习

本文链接：https://blog.csdn.net/liudehua6/article/details/112858260

版权

目标检测专栏收录该内容

1 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何在ContourNet项目中生成可视化结果。作者提供了在word_eval.py文件中的具体修改步骤，包括输入图像路径和输出图像路径的设置，并提醒用户需要预先创建输出路径。代码示例展示了使用OpenCV进行轮廓描边的实现。

摘要由CSDN通过智能技术生成

论文：https://arxiv.org/abs/2004.04940

代码：https://github.com/wangyuxin87/ContourNet

源代码中有生成可视化结果的代码，查看issue时看到还是有些朋友不会设置，这里举个例子说明一下。

如果要生成可是化结果，可以在/ContourNet/maskrcnn_benchmark/data/datasets/evaluation/word/word_eval.py Line480左右

的地方做如下两处修改，一个是输入图像的路径，一个是输出图像的路径。

注意：输出路径需要自己创建好，代码中没有创建。

                # visualization
                if cfg.DATASETS.Test_Visual:
                    im_write = cv2.imread(      # 'path to ic15 test image (must same to paths_catalog.py configured)'
                         '/mnt/ContourNet/datasets/ic15/ic15_test_images/' + im_w_name)[
                               :, :, ::-1]
                    for i in keep:
                        box = esd[i]
                        # print(box)
                        # assert 1<0
                        box = np.array(box['seg_rorect'])
                        box = np.around(box).astype(np.int32)
                        cv2.polylines(im_write[:, :, ::-1], [box.astype(np.int32).reshape((-1, 1, 2))], True,
                                      color=(0, 255, 0), thickness=2)  # 0,255,255 y 0,255,0 g
                    cv2.imwrite('/mnt/ContourNet/datasets/ic15/ic15_test_result/' + im_w_name, im_write[:, :, ::-1])     # 'path to visualization'
            else:

Enki08

关注

0
点赞
踩
2

收藏

觉得还不错? 一键收藏
1
评论
ContourNet test IC15 数据集生成可视化结果

论文：https://arxiv.org/abs/2004.04940代码：https://github.com/wangyuxin87/ContourNet源代码中有生成可视化结果的代码，查看issue时看到还是有些朋友不会设置，这里举个例子说明一下。如果要生成可是化结果，可以在/ContourNet/maskrcnn_benchmark/data/datasets/evaluation/word/word_eval.py Line480左右的地方做如下两处修改，一个是输入图像的路径，一个
复制链接

扫一扫