Please help with the cnn method used in the av1_rd_pick_partition process, thanks

23 views
Skip to first unread message

annie xu

unread,
Oct 9, 2022, 3:08:47 AM10/9/22
to AV1 Discussion
There is some cnn based logic in the av1_rd_pick_partition process, what is the rationale for this part of the code? Are there any relevant papers or materials that I can study? Please help with recommendations, thanks.
In the course of reading the code, I found some functions that may be related to the cnn decision method, as follows:
av1_collect_motion_search_features_sb
av1_intra_mode_cnn_partition
av1_simple_motion_search_based_split
av1_simple_motion_search_prune_rect
av1_ml_early_term_after_split
av1_ml_prune_rect_partition
av1_ml_prune_ab_partition
av1_ml_prune_4_partition
av1_ml_predict_breakout
So I would like to learn the theoretical process of these codes, thanks

Looking forward to the reply, thanks!

Cheng Chen

unread,
Oct 10, 2022, 2:47:42 AM10/10/22
to av1-d...@aomedia.org
This is the paper you are looking for (although for vp9, the idea is similar):

Machine Learning Accelerated Partition Search for Video Encoding


--
You received this message because you are subscribed to the Google Groups "AV1 Discussion" group.
To unsubscribe from this group and stop receiving emails from it, send an email to av1-discuss...@aomedia.org.
To view this discussion on the web visit https://groups.google.com/a/aomedia.org/d/msgid/av1-discuss/05bb7824-2d65-4b8c-b683-b9b371b81f87n%40aomedia.org.

annie xu

unread,
Oct 11, 2022, 12:42:07 AM10/11/22
to AV1 Discussion, chen...@google.com
Thank you very much for your reply.
Reply all
Reply to author
Forward
0 new messages