Nenhuma descrição

xue50 78acc468a5 修改box问题,回退到之前版本 7 meses atrás
config 0f7ce4ab80 修复tensorboard绘line图重叠bug 10 meses atrás
lcnn ac4676ffde 重构Line_net2 10 meses atrás
libs 6be97eaaa7 merge with dev 10 meses atrás
models 78acc468a5 修改box问题,回退到之前版本 7 meses atrás
tools ac4676ffde 重构Line_net2 10 meses atrás
utils 2c1a9b7db1 tensorboard 加入lmap jmap 8 meses atrás
.gitignore 14a3221d39 修改model.predict 7 meses atrás
aaa.py c2cffb6b1a predict_box内无线段时,选box内点组成线段最长的 两个点组成的线段返回 8 meses atrás
readme.md 1e96c80b34 完善readme.md,发布版本 8 meses atrás
train——line_rcnn.py 0f7ce4ab80 修复tensorboard绘line图重叠bug 10 meses atrás
冻结参数训练.py 02ffb35cac faster_rcnn and lcnn backbone is OK 11 meses atrás

readme.md

MultiVisionModels Platform

1.Dependences

Name Version Build Channel
python 3.12.8 h3f84c4b_1_cpython conda-forge
pytorch 2.3.1 py3.12_cuda12.1_cudnn8_0 pytorch
pytorch-cuda 12.1 hde6ce7c_6 pytorch
torchvision 0.17.2 pypi_0 pypi
numpy 1.26.3 py312h8753938_0 conda-forge
matplotlib 3.10.0 pypi_0 pypi

A100(40G) train edition

2.Overview

Include objection dectection ,keypoint detection,instance segment detection and line dectection.

网络结构改变:

相较与原LCNN,不再经过line_head层,直接使用原始的feature的前5层作为head;

在line_predictor,使用lmap替换x进行双线性插值提取线段特征

predict2.py文件 添加功能:box内无线段时,选box内点组成线段最长的 两个点组成的线段返回

(酸再生工件,会出现长线穿过,识别不出短线 这个在工件拍全时效果很好,拍不全会影响结果)