|
马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
因学习以及作业,现复现一个CVPR2017的一个手势识别图像处理项目
下载源代码以及数据集后运行训练代码,始终报错
我感觉路径是弄对了的
- from __future__ import print_function, unicode_literals
- import tensorflow as tf
- import os
- import sys
- from nets.ColorHandPose3DNetwork import ColorHandPose3DNetwork
- from data.BinaryDbReader import BinaryDbReader
- from utils.general import LearningRateScheduler, load_weights_from_snapshot
- # training parameters
- train_para = {'lr': [1e-5, 1e-6, 1e-7],
- 'lr_iter': [20000, 30000],
- 'max_iter': 40000,
- 'show_loss_freq': 1000,
- 'snapshot_freq': 5000,
- 'snapshot_dir': 'snapshots_handsegnet'}
- # get dataset
- dataset = BinaryDbReader(mode='training',
- batch_size=8, shuffle=True,
- hue_aug=True, random_crop_to_size=True)
- # build network graph
- data = dataset.get()
复制代码
- self.path_to_db = './date/bin/'
- self.num_samples = 0
- if mode == 'training':
- self.path_to_db += 'rhd_training.bin'
- self.num_samples = 41258
- elif mode == 'evaluation':
- self.path_to_db += 'rhd_evaluation.bin'
- self.num_samples = 2728
- else:
- assert 0, "Unknown dataset mode."
- assert os.path.exists(self.path_to_db), "Could not find the binary data file!"
复制代码
error:- Traceback (most recent call last):
- File "hand3d-master/training_handsegnet.py", line 25, in <module>
- hue_aug=True, random_crop_to_size=True)
- File "hand3d-master\data\BinaryDbReader.py", line 53, in __init__
- assert os.path.exists(self.path_to_db), "Could not find the binary data file!"
- AssertionError: Could not find the binary data file!
复制代码
|
|