diff --git a/requirement.txt b/requirement.txt new file mode 100644 index 0000000..14b70e2 --- /dev/null +++ b/requirement.txt @@ -0,0 +1,151 @@ +accelerate==1.8.1 +aiohappyeyeballs==2.6.1 +aiohttp==3.12.14 +aiosignal==1.4.0 +airportsdata==20250706 +annotated-types==0.7.0 +anyio==4.9.0 +astor==0.8.1 +async-timeout==5.0.1 +attrs==25.3.0 +bitsandbytes==0.46.1 +blake3==1.0.5 +cachetools==6.1.0 +certifi==2025.6.15 +charset-normalizer==3.4.2 +click==8.2.1 +cloudpickle==3.1.1 +compressed-tensors==0.10.2 +cupy-cuda12x==13.5.1 +depyf==0.18.0 +dill==0.4.0 +diskcache==5.6.3 +distro==1.9.0 +dnspython==2.7.0 +einops==0.8.1 +email_validator==2.2.0 +exceptiongroup==1.3.0 +fastapi==0.116.1 +fastapi-cli==0.0.8 +fastapi-cloud-cli==0.1.4 +fastrlock==0.8.3 +filelock==3.18.0 +frozenlist==1.7.0 +fsspec==2025.5.1 +gguf==0.17.1 +h11==0.16.0 +hf-xet==1.1.5 +httpcore==1.0.9 +httptools==0.6.4 +httpx==0.28.1 +huggingface-hub==0.33.2 +idna==3.10 +interegular==0.3.3 +itsdangerous==2.2.0 +Jinja2==3.1.6 +jiter==0.10.0 +jsonschema==4.24.0 +jsonschema-specifications==2025.4.1 +lark==1.2.2 +llguidance==0.7.30 +llvmlite==0.44.0 +lm-format-enforcer==0.10.11 +markdown-it-py==3.0.0 +MarkupSafe==2.1.5 +mdurl==0.1.2 +mistral_common==1.7.0 +mpmath==1.3.0 +msgpack==1.1.1 +msgspec==0.19.0 +multidict==6.6.3 +nest-asyncio==1.6.0 +networkx==3.4.2 +ninja==1.11.1.4 +numba==0.61.2 +numpy==2.1.2 +nvidia-cublas-cu12==12.6.4.1 +nvidia-cuda-cupti-cu12==12.6.80 +nvidia-cuda-nvrtc-cu12==12.6.77 +nvidia-cuda-runtime-cu12==12.6.77 +nvidia-cudnn-cu12==9.5.1.17 +nvidia-cufft-cu12==11.3.0.4 +nvidia-cufile-cu12==1.11.1.6 +nvidia-curand-cu12==10.3.7.77 +nvidia-cusolver-cu12==11.7.1.2 +nvidia-cusparse-cu12==12.5.4.2 +nvidia-cusparselt-cu12==0.6.3 +nvidia-ml-py==12.575.51 +nvidia-nccl-cu12==2.26.2 +nvidia-nvjitlink-cu12==12.6.85 +nvidia-nvshmem-cu12==3.3.9 +nvidia-nvtx-cu12==12.6.77 +openai==1.90.0 +opencv-python-headless==4.12.0.88 +orjson==3.11.0 +outlines==0.1.11 +outlines_core==0.1.26 +packaging==25.0 +pandas==2.3.1 +partial-json-parser==0.2.1.1.post6 +pillow==11.2.1 +prometheus-fastapi-instrumentator==7.1.0 +prometheus_client==0.22.1 +propcache==0.3.2 +protobuf==6.31.1 +psutil==7.0.0 +py-cpuinfo==9.0.0 +pybase64==1.4.1 +pycountry==24.6.1 +pydantic==2.11.7 +pydantic-extra-types==2.10.5 +pydantic-settings==2.10.1 +pydantic_core==2.33.2 +Pygments==2.19.2 +pynvml==12.0.0 +python-dateutil==2.9.0.post0 +python-dotenv==1.1.1 +python-json-logger==3.3.0 +python-multipart==0.0.20 +pytorch-triton==3.3.1+gitc8757738 +pytz==2025.2 +PyYAML==6.0.2 +pyzmq==27.0.0 +ray==2.47.1 +referencing==0.36.2 +regex==2024.11.6 +requests==2.32.4 +rich==14.0.0 +rich-toolkit==0.14.8 +rignore==0.6.2 +rpds-py==0.26.0 +safetensors==0.5.3 +scipy==1.15.3 +sentencepiece==0.2.0 +sentry-sdk==2.32.0 +shellingham==1.5.4 +six==1.17.0 +sniffio==1.3.1 +starlette==0.47.1 +sympy==1.14.0 +tiktoken==0.9.0 +tokenizers==0.21.2 +torch==2.7.0 +torchaudio==2.7.0 +torchvision==0.22.0 +tqdm==4.67.1 +transformers==4.53.1 +triton==3.3.0 +typer==0.16.0 +typing-inspection==0.4.1 +typing_extensions==4.14.0 +tzdata==2025.2 +ujson==5.10.0 +urllib3==2.5.0 +uvicorn==0.35.0 +uvloop==0.21.0 +vllm==0.9.2 +watchfiles==1.1.0 +websockets==15.0.1 +xformers==0.0.30 +xgrammar==0.1.19 +yarl==1.20.1 diff --git a/sishu/best.onnx b/sishu/best.onnx new file mode 100644 index 0000000..ad1d7e0 Binary files /dev/null and b/sishu/best.onnx differ diff --git a/sishu/best_fixed_dim.onnx b/sishu/best_fixed_dim.onnx new file mode 100644 index 0000000..ad1d7e0 Binary files /dev/null and b/sishu/best_fixed_dim.onnx differ diff --git a/sishu/runs/detect/predict/test.jpg b/sishu/runs/detect/predict/test.jpg new file mode 100644 index 0000000..92598d8 Binary files /dev/null and b/sishu/runs/detect/predict/test.jpg differ diff --git a/sishu/runs/detect/predict2/test.jpg b/sishu/runs/detect/predict2/test.jpg new file mode 100644 index 0000000..92598d8 Binary files /dev/null and b/sishu/runs/detect/predict2/test.jpg differ diff --git a/sishu/test.jpg b/sishu/test.jpg new file mode 100644 index 0000000..0673ea5 Binary files /dev/null and b/sishu/test.jpg differ