TF-cpn人体姿态估计运行mptest.py"No module named 'lib_kernel.lib_nms.gpu_nms'"解决方案

这个BUG解决了整整一天，特此纪念，记于马鞍山软件园

原工程链接https://github.com/chenyilun95/tf-cpn/
训练完毕后，执行python mptest.py -d 0-1 -m log/model_dump/snapshot_18.ckpt
报错

在这里说明一下，执行该命令前，已经完成lib.kernel/lib_nms下的compile.sh编译工作，编译就不多说了，首先手动删除两个.so文件，然后打开终端sh compile.sh

出现错误后各种百度各种谷歌，原工程下也有人问类似的问题，链接https://github.com/chenyilun95/tf-cpn/issues/56

但是尝试了，依旧无法解决
于是在上图文件夹下打开终端直接import gpu_nms,发现可以import

但是mptest.py里为什么就不能导入呢，
mptest.py import代码段如下:

import os
import os.path as osp
import numpy as np
import argparse
from config import cfg
import cv2
import sys
import time

import tensorflow as tf

from tfflat.base import Tester
from tfflat.utils import mem_info
from tfflat.logger import colorlogger
from network import Network

from lib_kernel.lib_nms.gpu_nms import gpu_nms
from lib_kernel.lib_nms.cpu_nms import cpu_soft_nms
from dataset import Preprocessing
from COCOAllJoints import COCOJoints

from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
from pycocotools import mask as COCOmask

怀疑代码附近是不是有空格，仔细检查后，没有，百思不得其解，而且难以解释为什么文件夹下可以import
在外卖送来之前已经打算放弃这个project
终于灵光一现。。。。。
把报错的直接放在最前面import

from lib_kernel.lib_nms.gpu_nms import gpu_nms
from lib_kernel.lib_nms.cpu_nms import cpu_soft_nms
import os
import os.path as osp
import numpy as np
import argparse
from config import cfg
import cv2
import sys
import time

import tensorflow as tf

from tfflat.base import Tester
from tfflat.utils import mem_info
from tfflat.logger import colorlogger
from network import Network

#from lib_kernel.lib_nms.gpu_nms import gpu_nms
#from lib_kernel.lib_nms.cpu_nms import cpu_soft_nms
from dataset import Preprocessing
from COCOAllJoints import COCOJoints

from pycocotools.coco import COCO
from pycocotools.cocoeval import COCOeval
from pycocotools import mask as COCOmask

终端再次运行

AP 和recall的结果一会也出来了

个人怀疑是不是前面的import有冲突，如果有大神知道原因麻烦评论告知

终于体会到科研的路多么艰辛，一个BUG能让我上瘾，还好我没放弃