thibo73800 / capsnet-traffic-sign-classifier

A Tensorflow implementation of CapsNet(Capsules Net) apply on german traffic sign dataset
Apache License 2.0
177 stars 90 forks source link

ValueError: unsupported pickle protocol: 4 #2

Closed jhasuman closed 7 years ago

jhasuman commented 7 years ago

Hi, I was trying to run train.py dataset on AWS cloud with ubuntu 14.04 but got this error, I am pasting all the errors, please help me help me on how to fix this ?

ValueErrorTraceback (most recent call last) /home/ubuntu/capsnet-traffic-sign-classifier/train.py in () 123 if name == 'main': 124 arguments = docopt(doc) --> 125 train(arguments[""], arguments["--ckpt"], arguments[""])

/home/ubuntu/capsnet-traffic-sign-classifier/train.py in train(dataset, ckpt, output) 53 return np.array(img) / 255 54 ---> 55 X_train, y_train, X_valid, y_valid, X_test, y_test = get_data(dataset) 56 57 X_train = X_train / 255

/home/ubuntu/capsnet-traffic-sign-classifier/data_handler.pyc in get_data(folder) 21 22 with open(training_file, mode='rb') as f: ---> 23 train = pickle.load(f) 24 with open(validation_file, mode='rb') as f: 25 valid = pickle.load(f)

/usr/lib/python2.7/pickle.pyc in load(file) 1376 1377 def load(file): -> 1378 return Unpickler(file).load() 1379 1380 def loads(str):

/usr/lib/python2.7/pickle.pyc in load(self) 856 while 1: 857 key = read(1) --> 858 dispatchkey 859 except _Stop, stopinst: 860 return stopinst.value

/usr/lib/python2.7/pickle.pyc in load_proto(self) 884 proto = ord(self.read(1)) 885 if not 0 <= proto <= 2: --> 886 raise ValueError, "unsupported pickle protocol: %d" % proto 887 dispatch[PROTO] = load_proto 888

ValueError: unsupported pickle protocol: 4

thibo73800 commented 7 years ago

I forgot to precise it in the README. I will add it. You need python3.

thibo73800 commented 7 years ago

I just push to up the test accuracy to 96,97%

thibo73800 commented 7 years ago

97% now

jhasuman commented 7 years ago

WooW!! That's Amazing!! I am hoping you are going to increase it further.