Initial commit

1345fab2 · luopl · 1345fab2 · 1345fab2 · 1345fab2 · 1345fab2
Commit 1345fab2 authored Jun 27, 2024 by luopl
20 changed files
--- a/models/bricks/ops/cuda/ms_deform_im2col_cuda.cuh
+++ b/models/bricks/ops/cuda/ms_deform_im2col_cuda.cuh
--- a/models/bricks/position_encoding.py
+++ b/models/bricks/position_encoding.py
--- a/models/bricks/post_process.py
+++ b/models/bricks/post_process.py
--- a/models/bricks/salience_transformer.py
+++ b/models/bricks/salience_transformer.py
--- a/models/bricks/set_criterion.py
+++ b/models/bricks/set_criterion.py
--- a/models/detectors/__init__.py
+++ b/models/detectors/__init__.py
--- a/models/detectors/base_detector.py
+++ b/models/detectors/base_detector.py
--- a/models/detectors/salience_detr.py
+++ b/models/detectors/salience_detr.py
--- a/models/matcher/hungarian_matcher.py
+++ b/models/matcher/hungarian_matcher.py
--- a/models/necks/__init__.py
+++ b/models/necks/__init__.py
--- a/models/necks/channel_mapper.py
+++ b/models/necks/channel_mapper.py
--- a/models/necks/repnet.py
+++ b/models/necks/repnet.py
--- a/optimizer/param_dict.py
+++ b/optimizer/param_dict.py
--- a/requirements.txt
+++ b/requirements.txt
--- a/test.py
+++ b/test.py
--- a/tools/benchmark_model.py
+++ b/tools/benchmark_model.py
--- a/tools/pytorch2onnx.py
+++ b/tools/pytorch2onnx.py
--- a/tools/visualize_datasets.py
+++ b/tools/visualize_datasets.py
--- a/transforms/__init__.py
+++ b/transforms/__init__.py
+from .transforms import *
+from .autoaugment import *
+from .convert_coco_polys_to_mask import ConvertCocoPolysToMask
+from .simple_copy_paste import SimpleCopyPaste
--- a/transforms/_functional_pil.py
+++ b/transforms/_functional_pil.py