Hubch / cad_recognition

cad drawings Recognition matching system with GNN
MIT License
6 stars 2 forks source link

done except framework adaption #2

Closed AilvenLiu closed 2 months ago

AilvenLiu commented 2 months ago

Almost complete, the super-high-resulotion pdf2img cad matching solution by target matching and matrices following a single class detection model, with preprocess and postprocess resolve solution in project.

The full code of small gray image matching & classification resolve solution is also included in this merge request with consist of the construction of Triplet DataSet, complex and combined loss function, matrice indices, distillation training strategy, et.al..

The detect model related solution is excluded due to it can be found and established following ultralytics/yolov5 completely: we use a ordinary yolov5 with a few modification temporary. (To optimize it in the next stage.)

Finally, all the new-added is operated in the independent sub-directory 'det_matrices' that does not influence the others of master at all.