nwojke / cosine_metric_learning

Deep Cosine Metric Learning for Person Re-identification
GNU General Public License v3.0
586 stars 201 forks source link

Architecture Inconsistency Fix #107

Open westonCoder opened 2 years ago

westonCoder commented 2 years ago

change network to be consistent with deep_sort/tools/freeze_model.py. This change solves deep_sort issues: #174,#156,#96 and issues in this repo: #35,#14,#6,#33. this also makes the model compatible with TensorRT supported layers