Hi, @yangxue0827 , I faced a problem when I run your training scripts: python train1.py in your docker environment.
python train1.py
The error information like these:
Traceback (most recent call last):
File "train1.py", line 12, in
import tensorflow.contrib.slim as slim
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/tensorflow/init.py", line 24, in
from tensorflow.python import *
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/tensorflow/python/init.py", line 47, in
import numpy as np
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/numpy/init.py", line 142, in
from . import add_newdocs
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/numpy/add_newdocs.py", line 13, in
from numpy.lib import add_newdoc
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/numpy/lib/init.py", line 23, in
from .npyio import *
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/numpy/lib/npyio.py", line 14, in
from ._datasource import DataSource
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/numpy/lib/_datasource.py", line 220, in
_file_openers = _FileOpeners()
File "/root/R2CNN_FPN/local/lib/python2.7/site-packages/numpy/lib/_datasource.py", line 162, in init
self._file_openers = {None: io.open}
AttributeError: 'module' object has no attribute 'open'
please try python3 train1.py.
@yangxue0827 Ok, Thanks! By the way, could you add vgg16 support? Because the Res50 and Res101 have out-of-memory on my GPU device.
@yangxue0827 I tried the python3 run train1.py, I got error like this:
Traceback (most recent call last):
File "train1.py", line 21, in
from libs.fast_rcnn import build_fast_rcnn1
File "/home/hardDisk/research/workspace/R2CNN_FPN_Tensorflow/tools/../libs/fast_rcnn/build_fast_rcnn1.py", line 14, in
from libs.box_utils import nms_rotate
File "/home/hardDisk/research/workspace/R2CNN_FPN_Tensorflow/tools/../libs/box_utils/nms_rotate.py", line 12, in
from libs.box_utils.rotate_polygon_nms import rotate_gpu_nms
ImportError: dynamic module does not define init function (PyInit_rotate_polygon_nms)
yangxue0827
