import yaml ImportError: No module named yaml等问题解决

最新推荐文章于 2024-08-17 16:21:42 发布

cv.exp

最新推荐文章于 2024-08-17 16:21:42 发布

阅读量1.3w

点赞数 2

分类专栏： Deep Learning

本文链接：https://blog.csdn.net/forest_world/article/details/53814753

版权

Deep Learning 专栏收录该内容

172 篇文章 6 订阅

订阅专栏

转载请注明：http://blog.csdn.net/forest_world

1、问题：

'TEST': {'BBOX_REG': True,
          'HAS_RPN': True,
          'MAX_SIZE': 1000,
          'NMS': 0.3,
          'PROPOSAL_METHOD': 'selective_search',
          'RPN_MIN_SIZE': 16,
          'RPN_NMS_THRESH': 0.7,
          'RPN_POST_NMS_TOP_N': 300,
          'RPN_PRE_NMS_TOP_N': 6000,
          'SCALES': [600],
          'SVM': False},
'TRAIN': {'ASPECT_GROUPING': True,
           'BATCH_SIZE': 128,
           'BBOX_INSIDE_WEIGHTS': [1.0, 1.0, 1.0, 1.0],
           'BBOX_NORMALIZE_MEANS': [0.0, 0.0, 0.0, 0.0],
           'BBOX_NORMALIZE_STDS': [0.1, 0.1, 0.2, 0.2],
           'BBOX_NORMALIZE_TARGETS': True,
           'BBOX_NORMALIZE_TARGETS_PRECOMPUTED': True,
           'BBOX_REG': True,
           'BBOX_THRESH': 0.5,
           'BG_THRESH_HI': 0.5,
           'BG_THRESH_LO': 0.0,
           'FG_FRACTION': 0.25,
           'FG_THRESH': 0.5,
           'HAS_RPN': True,
           'IMS_PER_BATCH': 1,
           'MAX_SIZE': 1000,
           'PROPOSAL_METHOD': 'gt',
           'RPN_BATCHSIZE': 256,
           'RPN_BBOX_INSIDE_WEIGHTS': [1.0, 1.0, 1.0, 1.0],
           'RPN_CLOBBER_POSITIVES': False,
           'RPN_FG_FRACTION': 0.5,
           'RPN_MIN_SIZE': 16,
           'RPN_NEGATIVE_OVERLAP': 0.3,
           'RPN_NMS_THRESH': 0.7,
           'RPN_POSITIVE_OVERLAP': 0.7,
           'RPN_POSITIVE_WEIGHT': -1.0,
           'RPN_POST_NMS_TOP_N': 2000,
           'RPN_PRE_NMS_TOP_N': 12000,
           'SCALES': [600],
           'SNAPSHOT_INFIX': '',
           'SNAPSHOT_ITERS': 10000,
           'USE_FLIPPED': True,
           'USE_PREFETCH': False},
'USE_GPU_NMS': True}
WARNING: Logging before InitGoogleLogging() is written to STDERR
F1222 15:20:26.013303 19220 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode.

解决：

2、问题：

caffe-fast-rcnn

解决：

    make -j8 && make pycaffe

3、问题：

hi@ubuntu:~/faster_rcnn/py-faster-rcnn$ ./tools/train_net.py --solver path/to/solver.prototxt --weights path/to/pretrain_model --imdb voc_2007_trainval --iters 100000 --cfg experiments/cfgs/faster_rcnn_end2end.yml
Called with args:
Namespace(cfg_file='experiments/cfgs/faster_rcnn_end2end.yml', gpu_id=0, imdb_name='voc_2007_trainval', max_iters=100000, pretrained_model='path/to/pretrain_model', randomize=False, set_cfgs=None, solver='path/to/solver.prototxt')
Traceback (most recent call last):
File "./tools/train_net.py", line 86, in <module>
cfg_from_file(args.cfg_file)
File "/home/hi/faster_rcnn/py-faster-rcnn/tools/../lib/fast_rcnn/config.py", line 259, in cfg_from_file
import yaml
ImportError: No module named yaml

解决：

hi@ubuntu:~/faster_rcnn/py-faster-rcnn$ sudo apt-get install python-yaml