"float" 对象在读取框架时无法解释为整数

Question

我正在尝试使用 imageio API 读取帧。我有一个 reader 作为我使用 imageio.get_reader(video_path,"ffmpeg") 收到的对象。我有以下框架 reader 函数

def read_frames(reader, frame_q, use_webcam):
    if use_webcam:
        time.sleep(15)
        frame_cnt = 0
        while True:
            #if frame_cnt % 5 == 0:    
            #    ret, frame = reader.read()
            #    cur_img = frame[:,:,::-1]
            #    frame_q.put(cur_img)
            #else:
            #    ret, frame = reader.read()
            ret, frame = reader.read()
            cur_img = frame[:,:,::-1] # bgr to rgb from opencv reader
            frame_q.put(cur_img)
            if frame_q.qsize() > 100:
                time.sleep(1)
            else:
                time.sleep(DELAY/1000.)

            #print(cur_img.shape)

    else:
        #for cur_img in reader: # this is imageio reader, it uses rgb
        nframes = reader.get_length() **#getting error here**
        # if nframes == float('inf') or nframes ==float('-inf'):
        #     return float("nan")
        # return int(nframes)  

        for ii in range(nframes):
            while frame_q.qsize() > 500: # so that we dont use huge amounts of memory
                time.sleep(1)
            cur_img = reader.get_next_data()
            frame_q.put(cur_img)
            #shape = cur_img.shape
            #noisy_img = np.uint8(cur_img.astype(np.float) + np.random.randn(*shape) * 20)
            #frame_q.put(noisy_img)
            if ii % 100 == 0:
                print("%i / %i frames in queue" % (ii, nframes))
        print("All %i frames in queue" % (nframes))

回溯：

    Traceback (most recent call last):
  File "/home/prashantb/anaconda3/envs/demo/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap
    self.run()
  File "/home/prashantb/anaconda3/envs/demo/lib/python3.6/multiprocessing/process.py", line 93, in run
    self._target(*self._args, **self._kwargs)
  File "multiprocess_detect_actions.py", line 67, in read_frames
    for ii in range(nframes):
TypeError: 'float' object cannot be interpreted as an integer

我尝试使用代码中注释的 float('inf')，但在这种情况下它没有用。我还尝试将浮点值转换为整数，但它显示相同的错误。非常感谢您对此的建议。谢谢。

Answer 1

在 imageio 2.5.0 版的发行说明中，他们提到了对 ffmpeg 插件的一些更改：

"The reader of the ffmpeg plugin now always reports inf as the number of frames. Use reader.count_frames() to get the actual number, or estimate it from the fps and duration in the meta data."

https://imageio.readthedocs.io/en/stable/releasenotes.html#version-2-5-0-06-02-2019

这应该可以解决您的问题。

"float" 对象在读取框架时无法解释为整数

"float" object cannot be interpreted as an integer while reading the frame

python

python-imageio