def main(args): vid_proto = proto_load(args.vid_file) frame_to_det = load_frame_to_det(vid_proto, args.det_dir) if not os.path.isdir(args.save_dir): try: os.makedirs(args.save_dir) except: pass track_files = glob(osp.join(args.track_dir, '*.track*')) if len(track_files) == 0: print "Warning: {} has no tracks".format(args.track_dir) return for track_file in track_files: track_proto = proto_load(track_file) vid_name = vid_proto['video'] assert vid_name == track_proto['video'] cls_name = osp.basename(track_file).split('.')[1] cls_index = imagenet_vdet_class_idx[cls_name] # spatial max pooling score_proto = raw_dets_spatial_max_pooling( vid_proto, track_proto, frame_to_det, cls_index, overlap_thres=args.overlap_thres) # temporal max pooling temporal_max_score_proto = score_proto_temporal_maxpool( score_proto, args.window) # interpolation interpolated_score_proto = score_proto_interpolation( temporal_max_score_proto, vid_proto) # save score proto save_file = osp.join(args.save_dir, '{}.{}.score'.format(vid_name, cls_name)) if osp.isfile(save_file): print "{} already exists.".format(save_file) continue proto_dump(interpolated_score_proto, save_file) print "{} created.".format(save_file)
def main(args): vid_proto = proto_load(args.vid_file) frame_to_det = load_frame_to_det(vid_proto, args.det_dir) if not os.path.isdir(args.save_dir): try: os.makedirs(args.save_dir) except: pass track_files = glob(osp.join(args.track_dir, '*.track*')) if len(track_files) == 0: print "Warning: {} has no tracks".format(args.track_dir) return for track_file in track_files: track_proto = proto_load(track_file) vid_name = vid_proto['video'] assert vid_name == track_proto['video'] cls_name = osp.basename(track_file).split('.')[1] cls_index = imagenet_vdet_class_idx[cls_name] # spatial max pooling score_proto = raw_dets_spatial_max_pooling(vid_proto, track_proto, frame_to_det, cls_index, overlap_thres=args.overlap_thres) # temporal max pooling temporal_max_score_proto = score_proto_temporal_maxpool(score_proto, args.window) # interpolation interpolated_score_proto = score_proto_interpolation(temporal_max_score_proto, vid_proto) # save score proto save_file = osp.join(args.save_dir, '{}.{}.score'.format(vid_name, cls_name)) if osp.isfile(save_file): print "{} already exists.".format(save_file) continue proto_dump(interpolated_score_proto, save_file) print "{} created.".format(save_file)
track_proto = proto_load(track_file) vid_name = vid_proto['video'] assert vid_name == track_proto['video'] cls_name = osp.basename(track_file).split('.')[1] cls_index = imagenet_vdet_class_idx[cls_name] # spatial max pooling score_proto = dets_spatial_max_pooling( vid_proto, track_proto, det_proto, cls_index, overlap_thres=args.overlap_thres) # temporal max pooling temporal_max_score_proto = score_proto_temporal_maxpool( score_proto, args.window) # interpolation interpolated_score_proto = score_proto_interpolation( temporal_max_score_proto, vid_proto) # save score proto save_file = osp.join(args.save_dir, '{}.{}.score'.format(vid_name, cls_name)) if osp.isfile(save_file): print "{} already exists.".format(save_file) continue proto_dump(interpolated_score_proto, save_file) print "{} created.".format(save_file)
track_files = glob(osp.join(args.track_dir, '*.track*')) if len(track_files) == 0: print "Warning: {} has no tracks".format(args.track_dir) sys.exit(0) for track_file in track_files: track_proto = proto_load(track_file) vid_name = vid_proto['video'] assert vid_name == track_proto['video'] cls_name = osp.basename(track_file).split('.')[1] cls_index = imagenet_vdet_class_idx[cls_name] # spatial max pooling score_proto = dets_spatial_max_pooling(vid_proto, track_proto, det_proto, cls_index, overlap_thres=args.overlap_thres) # temporal max pooling temporal_max_score_proto = score_proto_temporal_maxpool(score_proto, args.window) # interpolation interpolated_score_proto = score_proto_interpolation(temporal_max_score_proto, vid_proto) # save score proto save_file = osp.join(args.save_dir, '{}.{}.score'.format(vid_name, cls_name)) if osp.isfile(save_file): print "{} already exists.".format(save_file) continue proto_dump(interpolated_score_proto, save_file) print "{} created.".format(save_file)