流程按照这个链接,视频讲解,提供所有文件,简单易懂传送门,里面有个公众号,里面也是干货满满.
我常用的是ubuntu系统,也不想再创建虚拟环境了,有一点点区别.
1)在anaconda目录下(lib/python3.7/site-packages)创建.pth文件,里面的路径设置还是一样.
2)训练过程中报错,如下:
我在使用tensorflow时这个错误我经常遇到,应该是tensorflow对gpu的使用,默认是马力全开的,然后就报内存了(我自己的手提没遇到过,奇怪),所以需要设置下gpu显存.在research/object_detection目录下修改model_main.py文件,屏蔽原来的config设置语句,设置限制显存使用率为0.8.然后就能完美训练了.
def main(unused_argv):
flags.mark_flag_as_required('model_dir')
flags.mark_flag_as_required('pipeline_config_path')
session_config = tf.ConfigProto(log_device_placement=True, allow_soft_placement=True)
session_config.gpu_options.per_process_gpu_memory_fraction = 0.8
config = tf.estimator.RunConfig(session_config=session_config,model_dir=FLAGS.model_dir)
#config = tf.estimator.RunConfig(model_dir=FLAGS.model_dir)