zhaoyinghan
774a29460c
fix ends loss
|
il y a 3 semaines | |
|---|---|---|
| config | il y a 10 mois | |
| lcnn | il y a 1 mois | |
| libs | il y a 1 mois | |
| models | il y a 3 semaines | |
| tools | il y a 1 mois | |
| utils | il y a 3 semaines | |
| .gitignore | il y a 6 mois | |
| aaa.py | il y a 8 mois | |
| d_data_spliter.py | il y a 1 mois | |
| readme.md | il y a 5 mois | |
| train——line_rcnn.py | il y a 5 mois | |
| 冻结参数训练.py | il y a 10 mois |
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.18.1 | pypi_0 | pypi |
| numpy | 2.2.25 | py312h8753938_0 | conda-forge |
| matplotlib | 3.10.0 | pypi_0 | pypi |
| opencv | 4.10.0 |
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内点组成线段最长的 两个点组成的线段返回
（酸再生工件，会出现长线穿过，识别不出短线 这个在工件拍全时效果很好，拍不全会影响结果）