Tangledup-ai/sam3_local
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fbcode/deeplearning/projects/sam3_release/sam3/train/data

Browse Source 
Reviewed By: JuanBesa

Differential Revision: D91210167

fbshipit-source-id: a563232f4bc82f6f3b99e53df1c88cf0f39747bb
This commit is contained in:
generatedunixname537391475639613
2026-02-03 13:14:44 -08:00
committed by meta-codesync[bot]
parent 99d02f28c8
commit 2ec3c0711a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sam3/train/data/collator.py
View File

@@ -194,7 +194,7 @@ def collate_fn_api(
offset_img_id = 0 offset_img_id = 0
offset_query_id = [0 for _ in range(num_stages)] offset_query_id = [0 for _ in range(num_stages)]
for i, data in enumerate(batch): for data in batch:
img_batch.extend([img.data for img in data.images]) img_batch.extend([img.data for img in data.images])
if data.raw_images is not None: if data.raw_images is not None:
@@ -209,7 +209,7 @@ def collate_fn_api(
datapoint_query_id_2_stage_query_id.append(offset_query_id[stage_id]) datapoint_query_id_2_stage_query_id.append(offset_query_id[stage_id])
offset_query_id[stage_id] += 1 offset_query_id[stage_id] += 1
for j, q in enumerate(data.find_queries): for q in data.find_queries:
stage_id = q.query_processing_order stage_id = q.query_processing_order
stages[stage_id].img_ids.append(q.image_id + offset_img_id) stages[stage_id].img_ids.append(q.image_id + offset_img_id)
if q.query_text not in text_batch: if q.query_text not in text_batch: