def inception_feature_for_equirectangular()

in FIDCalculations.py [0:0]

• 14 lines of code
• 6 McCabe index (conditional complexity)

def inception_feature_for_equirectangular(equirectangular_images,incp_img_size,incp_model):
    # standarize eq_im to batch,h,w,channels for ease
    if len(np.shape(equirectangular_images))<4:
        equirectangular_images=np.expand_dims(equirectangular_images,axis=0)
    batch,h,w,c=np.shape(equirectangular_images)
    all_faces=[]
    for b in range(0,batch):
        cube_faces=py360convert.e2c(equirectangular_images[b],cube_format='list',face_w=incp_img_size)
        # here we ignore top and bottom since they usually do not have much features
        cube_faces=cube_faces[0:4]
        if b==0:
            all_faces=np.squeeze(cube_faces)
        else:
            all_faces=np.append(all_faces,cube_faces,axis=0)

    inception_features=incp_model.predict(all_faces)
    return inception_features