FEAT：LineMod处理完成

lib/ARIAnnotation.py → ARI/ARIAnnotation.py

@@ -1,5 +1,5 @@
 import cv2
-from utils.TAUUtils import draw2DBBOX, draw3DBBOX 
+from TAU.TAUUtils import draw2DBBOX, draw3DBBOX 
 
 class ARIAnnotation2D(object):
 

lib/ARICOCO.py → ARI/ARICOCO.py

@@ -1,5 +1,5 @@
 
-from lib.ARIAnnotation import ARIDataset
+from ARI.ARIAnnotation import ARIDataset
 
 
 class ARICOCO(object):

lib/Linemod.py → ARI/Linemod.py

@@ -1,11 +1,9 @@
 import os
 
-from django.forms import models
+from ARI.ARIAnnotation import ARIAnnotation2D, ARIDataset, ARIImage
+from TAU.TAUUtils import *
 
-from lib.ARIAnnotation import ARIAnnotation2D, ARIDataset, ARIImage
-from utils.TAUUtils import *
-
-class ARILinemod(ARIDataset):
+class Linemod(ARIDataset):
     intrinsic = np.array([[572.4114, 0., 325.2611],
                           [0., 573.57043, 242.04899],
                           [0., 0., 1.]])

ARI/__init__.py

@@ -0,0 +1 @@
+from .Linemod import Linemod

utils/TAUUtils.py → TAU/TAUUtils.py

@@ -25,7 +25,7 @@ def calculateKeyPoint2D(points3D, intrinsic, RT):
     tmp = np.matmul(np.matmul(intrinsic, RT_np), points3D_np)
     tmp[0, :] = tmp[0, :] / tmp[2, :]
     tmp[1, :] = tmp[1, :] / tmp[2, :]
-    res = tmp[:2, :].flatten()
+    res = tmp[:2, :].T.flatten()
     return list(res)
 
 ## draw bbox

__init__.py

@@ -1 +1 @@
-from .utils import TAUUtils
+from .TAU import TAUUtils

main.py

@@ -1,7 +1,7 @@
 from cv2 import line
-from lib.Linemod import ARILinemod
+from ARI.Linemod import Linemod
 
 if __name__ == "__main__":
     sourceDir = "E:\code\PoseFuse\data\Linemod_preprocessed"
-    linemod = ARILinemod(sourceDir, "linemod")
+    linemod = Linemod(sourceDir, "linemod")
     linemod.images[1].visBBOX3D()