dsgou / annotator Goto Github PK

Now that the csvs can be read, the tool cannot open .bag files, due to an .avi conversion error. Changing some lines in the rosbagRGB.py file, the tool manages to write the .avi file, but cannot read it!!

Line 118 and forward, were changed to:

height, width, bytesPerComponent, tmp = np.array(image_buffer[0]).shape

video_writer = cv2.VideoWriter(rgbFileName, fourcc, framerate, (width,height), cv2.IMREAD_COLOR)

if video_writer.isOpened():
	result = True
	i= 0

	image_buffer = np.delete(np.array(image_buffer[0]),-1,1)

Please take a look at the new .avi reading method, since the problem should be creeping around there!

With lots of love,

George

Whenever I open a rosbag and then try to open the corresponding .csv file (that was created with the tool and not manually) the program throws an exception. Looking into the code, the problem should be in the video/rosbagRGB.py file, in the buffer_video_csv function.