diff --git a/dataset b/dataset index df1324d..3393678 100644 --- a/dataset +++ b/dataset @@ -7,6 +7,8 @@ from deepgtav.client import Client import argparse import time import cv2 +import pandas as pd +import random # Stores a dataset file with data coming from DeepGTAV if __name__ == '__main__': @@ -16,6 +18,15 @@ if __name__ == '__main__': parser.add_argument('-d', '--dataset_path', default='dataset.pz', help='Place to store the dataset') args = parser.parse_args() + # Get a random start location + df = pd.read_csv("gtav.rand.locations.csv", header=None) + randIdx = random.randint(0,349) + startLocation = [0,0,0] + startLocation = df.iloc[randIdx].tolist() + + # If you want a specific location specify it here: + # startLocation = [0,0,0] + # Creates a new connection to DeepGTAV using the specified ip and port. # If desired, a dataset path and compression level can be set to store in memory all the data received in a gziped pickle file. client = Client(ip=args.host, port=args.port, datasetPath=args.dataset_path, compressionLevel=9) @@ -24,7 +35,7 @@ if __name__ == '__main__': # See deepgtav/messages.py to see what options are supported dataset = Dataset(rate=30, frame=[320,160], throttle=True, brake=True, steering=True, vehicles=True, peds=True, reward=[15.0, 0.0], direction=None, speed=True, yawRate=True, location=True, time=True) # Send the Start request to DeepGTAV. - scenario = Scenario(drivingMode=[786603,15.0]) # Driving style is set to normal, with a speed of 15.0 mph. All other scenario options are random. + scenario = Scenario(location=startLocation, drivingMode=[786603,15.0]) # Driving style is set to normal, with a speed of 15.0 mph. All other scenario options are random. client.sendMessage(Start(dataset=dataset,scenario=scenario)) # Start listening for messages coming from DeepGTAV. We do it for 80 hours