马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
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!
|