Refactor project structure

modules/ANSMOT/ByteTrackNCNN/src/NCNNRect.cpp

@@ -0,0 +1,147 @@
+#include <algorithm>
+#include "NCNNRect.h"
+template <typename T>
+ByteTrackNCNN::Rect<T>::Rect(const T& x, const T& y, const T& width, const T& height) :
+	tlwh({ x, y, width, height })
+{
+}
+
+template <typename T>
+ByteTrackNCNN::Rect<T>::~Rect()
+{
+}
+
+template <typename T>
+const T& ByteTrackNCNN::Rect<T>::x() const
+{
+	return tlwh[0];
+}
+
+template <typename T>
+const T& ByteTrackNCNN::Rect<T>::y() const
+{
+	return tlwh[1];
+}
+
+template <typename T>
+const T& ByteTrackNCNN::Rect<T>::width() const
+{
+	return tlwh[2];
+}
+
+template <typename T>
+const T& ByteTrackNCNN::Rect<T>::height() const
+{
+	return tlwh[3];
+}
+
+template <typename T>
+T& ByteTrackNCNN::Rect<T>::x()
+{
+	return tlwh[0];
+}
+
+template <typename T>
+T& ByteTrackNCNN::Rect<T>::y()
+{
+	return tlwh[1];
+}
+
+template <typename T>
+T& ByteTrackNCNN::Rect<T>::width()
+{
+	return tlwh[2];
+}
+
+template <typename T>
+T& ByteTrackNCNN::Rect<T>::height()
+{
+	return tlwh[3];
+}
+
+template <typename T>
+const T& ByteTrackNCNN::Rect<T>::tl_x() const
+{
+	return tlwh[0];
+}
+
+template <typename T>
+const T& ByteTrackNCNN::Rect<T>::tl_y() const
+{
+	return tlwh[1];
+}
+
+template <typename T>
+T ByteTrackNCNN::Rect<T>::br_x() const
+{
+	return tlwh[0] + tlwh[2];
+}
+
+template <typename T>
+T ByteTrackNCNN::Rect<T>::br_y() const
+{
+	return tlwh[1] + tlwh[3];
+}
+
+template <typename T>
+ByteTrackNCNN::Tlbr<T> ByteTrackNCNN::Rect<T>::getTlbr() const
+{
+	return {
+		tlwh[0],
+		tlwh[1],
+		tlwh[0] + tlwh[2],
+		tlwh[1] + tlwh[3],
+	};
+}
+
+template <typename T>
+ByteTrackNCNN::Xyah<T> ByteTrackNCNN::Rect<T>::getXyah() const
+{
+	return {
+		tlwh[0] + tlwh[2] / 2,
+		tlwh[1] + tlwh[3] / 2,
+		tlwh[2] / tlwh[3],
+		tlwh[3],
+	};
+}
+
+template<typename T>
+float ByteTrackNCNN::Rect<T>::calcIoU(const Rect<T>& other) const
+{
+	const float box_area = (other.tlwh[2] + 1) * (other.tlwh[3] + 1);
+	const float iw = std::min(tlwh[0] + tlwh[2], other.tlwh[0] + other.tlwh[2]) - std::max(tlwh[0], other.tlwh[0]) + 1;
+	float iou = 0;
+	if (iw > 0)
+	{
+		const float ih = std::min(tlwh[1] + tlwh[3], other.tlwh[1] + other.tlwh[3]) - std::max(tlwh[1], other.tlwh[1]) + 1;
+		if (ih > 0)
+		{
+			const float ua = (tlwh[0] + tlwh[2] - tlwh[0] + 1) * (tlwh[1] + tlwh[3] - tlwh[1] + 1) + box_area - iw * ih;
+			iou = iw * ih / ua;
+		}
+	}
+	return iou;
+}
+
+template<typename T>
+ByteTrackNCNN::Rect<T> generate_rect_by_tlbr(const ByteTrackNCNN::Tlbr<T>& tlbr)
+{
+	return ByteTrackNCNN::Rect<T>(tlbr[0], tlbr[1], tlbr[2] - tlbr[0], tlbr[3] - tlbr[1]);
+}
+
+template<typename T>
+ByteTrackNCNN::Rect<T> generate_rect_by_xyah(const ByteTrackNCNN::Xyah<T>& xyah)
+{
+	const auto width = xyah[2] * xyah[3];
+	return ByteTrackNCNN::Rect<T>(xyah[0] - width / 2, xyah[1] - xyah[3] / 2, width, xyah[3]);
+}
+
+// explicit instantiation
+template class ByteTrackNCNN::Rect<int>;
+template class ByteTrackNCNN::Rect<float>;
+
+template ByteTrackNCNN::Rect<int> ByteTrackNCNN::generate_rect_by_tlbr<int>(const ByteTrackNCNN::Tlbr<int>&);
+template ByteTrackNCNN::Rect<float> ByteTrackNCNN::generate_rect_by_tlbr<float>(const ByteTrackNCNN::Tlbr<float>&);
+
+template ByteTrackNCNN::Rect<int> ByteTrackNCNN::generate_rect_by_xyah<int>(const ByteTrackNCNN::Xyah<int>&);
+template ByteTrackNCNN::Rect<float> ByteTrackNCNN::generate_rect_by_xyah<float>(const ByteTrackNCNN::Xyah<float>&);