该楼层疑似违规已被系统折叠 隐藏此楼查看此楼

python新手，做毕设用到，相关程序如下

file_num = g_file_num + int(

filename[filename.find("_")+1:filename.find(".")])

这一行报错：

Traceback (most recent call last):

File "F:\MATLAB\preprocess_cad60.py", line 118, in

extract_images_and_matricize_keypoints()

File "F:\MATLAB\preprocess_cad60.py", line 68, in extract_images_and_matricize_keypoints

filename[filename.find("_")+1:filename.find(".")])

ValueError: invalid literal for int() with base 10: 'activityLabel'

就上面这部分。非常感谢大佬的帮助！！

如果需要看全部代码的话：

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import os

import numpy as np

import pickle as pkl

import scipy.io as sio

import itertools

def remove_orientations_and_make_2d(joint_row):

"""As explained, this function removes all the orientation information and

makes each 3d pose 2d. It will also remove all the confidence bits from the

array"""

joint_row_without_orientations = []

for i in range(11):

del joint_row[:10]

joint_row_without_orientations.extend(joint_row[:3])

del joint_row[:4]

for i in range(4):

joint_row_without_orientations.extend(joint_row[:3])

del joint_row[:4]

joint_row_2d = []

joint_row_norm = []

for i in range(15):

x, y, z = joint_row_without_orientations[:3]

del joint_row_without_orientations[:3]

X = (156.8584456124928 + 0.0976862095248 * x - 0.0006444357104 * y

+ 0.0015715946682 * z)

Y = (125.5357201011431 + 0.0002153447766 * x - 0.1184874093530 * y

- 0.0022134485957 * z)

joint_row_2d.append([X,Y])

joint_row_norm.append([X/320, Y/240])#Normalized

return joint_row_2d, joint_row_norm

def extract_images_and_matricize_keypoints():

""" As the title suggests the following function will modify the title of

each file to match that of the folder in which it lies and then will also

matricize the keypoint file associated with it """

dat_dir = 'data/cad60_dataset'

try:

joints = pkl.load(open(dat_dir+'/joints.p', 'rb'))

except:

joints = []

mat_joints = []

subdir_list = []

for filename in os.listdir(dat_dir):

if os.path.isdir(dat_dir+'/'+filename):

subdir_list.append(filename)

print subdir_list

if not os.path.exists('data/cad60_dataset/images'):

os.mkdir('data/cad60_dataset/images')

g_file_num = 0

for subdir in subdir_list:

if not (subdir == 'images' or subdir == 'mark' or subdir == 'crop'

or subdir == 'joint'):

#Record the number of files (Depth OR RGB) in the folder

num_files_in_folder = int(len(os.listdir(dat_dir+'/'+subdir))/2)

print "Extracting files from dir: {}".format(subdir)

print "Number of files in the current folder:{}".format(

num_files_in_folder)

for filename in os.listdir(dat_dir+'/'+subdir):

file_num = g_file_num + int(

filename[filename.find("_")+1:filename.find(".")])

file_type = filename[0] #R(GB) or D

#Uncomment following line(and comment line below that one)

#if USING MULTIPLE DIRECTORIES from the CAD 60 dataset.

#os.rename(dat_dir+'/'+subdir+'/'+filename,

#dat_dir+'/images/'+subdir+file_num+'_'+file_type+'.jpg')

os.rename(dat_dir+'/'+subdir+'/'+filename,

dat_dir+'/images/'+str(file_num)+'_'+file_type+'.jpg')

print "Finished adding all files. Now extracting joint info"

with open(dat_dir+'/'+subdir+'.txt', 'rU') as txt_file:

lines_list = txt_file.read()

lines_list = lines_list.split('\n')[:-1]

for line in lines_list:

joint_row = line.split(',')[:-1]

if joint_row:

try:

joint_row[1:] = (

[float(joint_str) for joint_str in joint_row[1:]])

except ValueError:

print "[ERROR] Could not convert string to float"

print joint_row

return

joint_row_norm = list(joint_row)

joint_row[1:], joint_row_norm[1:] = (

remove_orientations_and_make_2d(joint_row[1:]))

#MATLAB Stuff

chain_mix = list(joint_row_norm)

chain_mix[0] = [g_file_num + int(chain_mix[0])]

chain = [item for sublist in chain_mix for item in sublist]

mat_joints.append(chain)

#Pickle Stuff

#joint_row[0] = subdir+joint_row[0]#UNCOMMENT if multiple

#directories of CAD60 dataset

joint_row[0] = str(chain_mix[0][0])#Since chain_mix[0] is a list

joints.append(joint_row)

else:

continue

print "Joint number is at {} = Number of files handled?".format(

chain_mix[0])

g_file_num += num_files_in_folder

os.rmdir(dat_dir+'/'+subdir)

print "Number of files handled: {}".format(g_file_num)

pkl.dump(joints, open(dat_dir+'/joints.p', 'wb'))

#sio.savemat(dat_dir+'/joints.mat', {'joints':mat_joints})#Store as mat

if __name__ == '__main__':

extract_images_and_matricize_keypoints()