E3Y55MPRKKDPTGI56RSA7YCGB33NSZYKHGCVHEUKRM2KJ2RNM5IQC RBXUHIO2WALXMYGFRWNDLVHFH4OOMCODI3K4HKZNWGWUX6INB6LAC NMQCXLNGPIKUKMNKEIVUBYUNJQDM63IVKRVMQHVV5TLH6OBNF3IAC NV7FXZ5QETWHE7EQHET5ZZUKH4UIAIRGQ42MR2IT5JCZDPRNEZRQC 3FAESP6NBZJHKN6VSGENUMZXPID6O5CGNQ3NHJC5Z3YXTAXJVWVAC U46LDPL7W76Z62KUTGA6QNB27C7HGBFUXT6E7V5X4ACTDUBTKAKAC EDYR5C55YKPEMJOS4O6YEUK5JYUWSX4NTPQGG4GLH5QL2O62GTPQC 2UBDFCJH2BG6U6SY2YDJ7QK4JOLUJAHOYZS3YRQ7E7U4UGP4YR5QC OFTU77S5FEE5SWAV7G4OWAQ7GY6LOEY65ZTMJCDMPMJGB7TWDIOQC glob_pattern = "*/*/[N,K]/*.png" #from SSD2/PrimaryDataset 7758648train("K1-5", 2, glob_pattern, true, 0.95, 64)
glob_pattern = "kiwi_set_*/*/[K]/*.png" #from SSD2/PrimaryDataset 7758648 hard coded a random selection of not, 1:1train("K1-7_R50", 20, glob_pattern, true, 0.90, 32)
glob_pattern = "kiwi_set_2023-11-13/*/[N,K]/*.png" #from SSD2/PrimaryDatasetmodel = "/media/david/SSD2/PrimaryDataset/model_K1-5_CPU_epoch-6-0.9795-2023-12-16.jld2"train("K1-5_ft1", 10, glob_pattern, model, 0.95, 64)
glob_pattern = "kiwi_set_2023-11-13/*/[N,K]/*.png" #from SSD2/PrimaryDatasetmodel = "/media/david/SSD2/PrimaryDataset/media/david/SSD2/PrimaryDataset/model_K1-6_CPU_epoch-3-0.9798-2024-02-01.jld2"train("K1-6_ft1", 10, glob_pattern, model, 0.95, 64)
fst = Metalhead.ResNet(18, pretrain = pretrain).layerslst = Flux.Chain(AdaptiveMeanPool((1, 1)), Flux.flatten, Dense(512 => classes))
fst = Metalhead.ResNet(50, pretrain = pretrain).layerslst = Flux.Chain(AdaptiveMeanPool((1, 1)), Flux.flatten, Dense(2048 => classes))
f = Metalhead.ResNet(18, pretrain = false).layersl = Flux.Chain(AdaptiveMeanPool((1, 1)), Flux.flatten, Dense(512 => model_classes))
f = Metalhead.ResNet(50, pretrain = false).layersl = Flux.Chain(AdaptiveMeanPool((1, 1)), Flux.flatten, Dense(2048 => model_classes))
model = "/media/david/SSD2/PrimaryDataset/model_K1-5_CPU_epoch-6-0.9795-2023-12-16.jld2"
model = "/media/david/SSD2/PrimaryDataset/model_K1-7_CPU_epoch-17-0.9126-2024-02-04.jld2"model = "/media/david/SSD2/PrimaryDataset/model_K1-7_R50_CPU_epoch-20-0.9136-2024-02-07.jld2"
f = Metalhead.ResNet(18, pretrain = false).layersl = Flux.Chain(AdaptiveMeanPool((1, 1)), Flux.flatten, Dense(512 => model_classes))
f = Metalhead.ResNet(50, pretrain = false).layersl = Flux.Chain(AdaptiveMeanPool((1, 1)), Flux.flatten, Dense(2048 => model_classes))
[deps.Accessors.extensions]AccessorsAxisKeysExt = "AxisKeys"AccessorsIntervalSetsExt = "IntervalSets"AccessorsStaticArraysExt = "StaticArrays"AccessorsStructArraysExt = "StructArrays"[deps.Accessors.weakdeps]AxisKeys = "94b1ba4f-4ee9-5380-92f1-94cde586c3c5"IntervalSets = "8197267c-284f-5f27-9208-e0e47529a953"Requires = "ae029012-a4dd-5104-9daa-d747884805df"StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"
[deps.ArrayInterface.extensions]ArrayInterfaceBandedMatricesExt = "BandedMatrices"ArrayInterfaceBlockBandedMatricesExt = "BlockBandedMatrices"ArrayInterfaceCUDAExt = "CUDA"ArrayInterfaceGPUArraysCoreExt = "GPUArraysCore"ArrayInterfaceStaticArraysCoreExt = "StaticArraysCore"ArrayInterfaceTrackerExt = "Tracker"[deps.ArrayInterface.weakdeps]BandedMatrices = "aae01518-5342-5314-be14-df237901396f"BlockBandedMatrices = "ffab5731-97b5-5995-9138-79e8c1846df0"CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"StaticArraysCore = "1e83bf80-4336-4d27-bf5d-d5a4f845583c"Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
[[deps.CPUSummary]]deps = ["CpuId", "IfElse", "PrecompileTools", "Static"]git-tree-sha1 = "601f7e7b3d36f18790e2caf83a882d88e9b71ff1"uuid = "2a0fbf3d-bb9c-48f3-b0a9-814d99fd7ab9"version = "0.2.4"
deps = ["AbstractFFTs", "Adapt", "BFloat16s", "CEnum", "CUDA_Driver_jll", "CUDA_Runtime_Discovery", "CUDA_Runtime_jll", "Crayons", "DataFrames", "ExprTools", "GPUArrays", "GPUCompiler", "KernelAbstractions", "LLVM", "LLVMLoopInfo", "LazyArtifacts", "Libdl", "LinearAlgebra", "Logging", "NVTX", "Preferences", "PrettyTables", "Printf", "Random", "Random123", "RandomNumbers", "Reexport", "Requires", "SparseArrays", "Statistics", "UnsafeAtomicsLLVM"]git-tree-sha1 = "95ac638373ac40e29c1b6d086a3698f5026ff6a6"
deps = ["AbstractFFTs", "Adapt", "BFloat16s", "CEnum", "CUDA_Driver_jll", "CUDA_Runtime_Discovery", "CUDA_Runtime_jll", "Crayons", "DataFrames", "ExprTools", "GPUArrays", "GPUCompiler", "KernelAbstractions", "LLVM", "LLVMLoopInfo", "LazyArtifacts", "Libdl", "LinearAlgebra", "Logging", "NVTX", "Preferences", "PrettyTables", "Printf", "Random", "Random123", "RandomNumbers", "Reexport", "Requires", "SparseArrays", "StaticArrays", "Statistics"]git-tree-sha1 = "baa8ea7a1ea63316fa3feb454635215773c9c845"
deps = ["ColorTypes", "FixedPointNumbers", "LinearAlgebra", "Requires", "Statistics", "TensorCore"]git-tree-sha1 = "a1f44953f2382ebb937d60dafbe2deea4bd23249"
deps = ["ColorTypes", "FixedPointNumbers", "LinearAlgebra", "SpecialFunctions", "Statistics", "TensorCore"]git-tree-sha1 = "600cc5508d66b78aae350f7accdb58763ac18589"
deps = ["Artifacts", "CompilerSupportLibraries_jll", "JLLWrappers", "Libdl", "Ogg_jll", "Pkg"]git-tree-sha1 = "f2c9884b0a97d3255988fbd8f062cafc6cb37b97"
deps = ["Artifacts", "CompilerSupportLibraries_jll", "JLLWrappers", "Libdl", "Ogg_jll"]git-tree-sha1 = "a4d6ed9ff6ae2080f15e588499b1324791a9265f"
[[deps.HistogramThresholding]]deps = ["ImageBase", "LinearAlgebra", "MappedArrays"]git-tree-sha1 = "7194dfbb2f8d945abdaf68fa9480a965d6661e69"uuid = "2c695a8d-9458-5d45-9878-1b8a99cf7853"version = "0.3.1"
version = "0.1.7"[[deps.ImageBinarization]]deps = ["HistogramThresholding", "ImageCore", "LinearAlgebra", "Polynomials", "Reexport", "Statistics"]git-tree-sha1 = "f5356e7203c4a9954962e3757c08033f2efe578a"uuid = "cbc4b850-ae4b-5111-9e64-df94c024a13d"version = "0.3.0"
version = "0.1.5"
deps = ["AbstractFFTs", "ColorVectorSpace", "Colors", "FixedPointNumbers", "MappedArrays", "MosaicViews", "OffsetArrays", "PaddedViews", "PrecompileTools", "Reexport"]git-tree-sha1 = "fc5d1d3443a124fde6e92d0260cd9e064eba69f8"
deps = ["AbstractFFTs", "ColorVectorSpace", "Colors", "FixedPointNumbers", "Graphics", "MappedArrays", "MosaicViews", "OffsetArrays", "PaddedViews", "Reexport"]git-tree-sha1 = "acf614720ef026d38400b3817614c45882d75500"
deps = ["DataStructures", "ImageCore", "LinearAlgebra", "LoopVectorization", "OffsetArrays", "Requires", "TiledIteration"]git-tree-sha1 = "6f0a801136cb9c229aebea0df296cdcd471dbcd1"
deps = ["ImageCore", "LinearAlgebra", "Requires", "TiledIteration"]git-tree-sha1 = "e7c68ab3df4a75511ba33fc5d8d9098007b579a8"
deps = ["AxisAlgorithms", "CoordinateTransformations", "ImageBase", "ImageCore", "Interpolations", "OffsetArrays", "Rotations", "StaticArrays"]git-tree-sha1 = "e0884bdf01bbbb111aea77c348368a86fb4b5ab6"
deps = ["AxisAlgorithms", "ColorVectorSpace", "CoordinateTransformations", "ImageBase", "ImageCore", "Interpolations", "OffsetArrays", "Rotations", "StaticArrays"]git-tree-sha1 = "8717482f4a2108c9358e5c3ca903d3a6113badc9"
deps = ["Base64", "FileIO", "Graphics", "ImageAxes", "ImageBase", "ImageBinarization", "ImageContrastAdjustment", "ImageCore", "ImageCorners", "ImageDistances", "ImageFiltering", "ImageIO", "ImageMagick", "ImageMetadata", "ImageMorphology", "ImageQualityIndexes", "ImageSegmentation", "ImageShow", "ImageTransformations", "IndirectArrays", "IntegralArrays", "Random", "Reexport", "SparseArrays", "StaticArrays", "Statistics", "StatsBase", "TiledIteration"]git-tree-sha1 = "d438268ed7a665f8322572be0dabda83634d5f45"
deps = ["Base64", "FileIO", "Graphics", "ImageAxes", "ImageBase", "ImageContrastAdjustment", "ImageCore", "ImageDistances", "ImageFiltering", "ImageIO", "ImageMagick", "ImageMetadata", "ImageMorphology", "ImageQualityIndexes", "ImageSegmentation", "ImageShow", "ImageTransformations", "IndirectArrays", "IntegralArrays", "Random", "Reexport", "SparseArrays", "StaticArrays", "Statistics", "StatsBase", "TiledIteration"]git-tree-sha1 = "5fa9f92e1e2918d9d1243b1131abe623cdf98be7"
deps = ["Adapt", "AxisAlgorithms", "ChainRulesCore", "LinearAlgebra", "OffsetArrays", "Random", "Ratios", "Requires", "SharedArrays", "SparseArrays", "StaticArrays", "WoodburyMatrices"]git-tree-sha1 = "88a101217d7cb38a7b481ccd50d21876e1d1b0e0"
deps = ["AxisAlgorithms", "ChainRulesCore", "LinearAlgebra", "OffsetArrays", "Random", "Ratios", "Requires", "SharedArrays", "SparseArrays", "StaticArrays", "WoodburyMatrices"]git-tree-sha1 = "00a19d6ab0cbdea2978fc23c5a6482e02c192501"
[[deps.LayoutPointers]]deps = ["ArrayInterface", "LinearAlgebra", "ManualMemory", "SIMDTypes", "Static", "StaticArrayInterface"]git-tree-sha1 = "62edfee3211981241b57ff1cedf4d74d79519277"uuid = "10f19ff3-798f-405d-979b-55457f8fc047"version = "0.1.15"
[[deps.LoopVectorization]]deps = ["ArrayInterface", "CPUSummary", "CloseOpenIntervals", "DocStringExtensions", "HostCPUFeatures", "IfElse", "LayoutPointers", "LinearAlgebra", "OffsetArrays", "PolyesterWeave", "PrecompileTools", "SIMDTypes", "SLEEFPirates", "Static", "StaticArrayInterface", "ThreadingUtilities", "UnPack", "VectorizationBase"]git-tree-sha1 = "0f5648fbae0d015e3abe5867bca2b362f67a5894"uuid = "bdcacae8-1622-11e9-2a5c-532679323890"version = "0.12.166"weakdeps = ["ChainRulesCore", "ForwardDiff", "SpecialFunctions"][deps.LoopVectorization.extensions]ForwardDiffExt = ["ChainRulesCore", "ForwardDiff"]SpecialFunctionsExt = "SpecialFunctions"
[[deps.PolyesterWeave]]deps = ["BitTwiddlingConvenienceFunctions", "CPUSummary", "IfElse", "Static", "ThreadingUtilities"]git-tree-sha1 = "240d7170f5ffdb285f9427b92333c3463bf65bf6"uuid = "1d0040c9-8b98-4ee7-8388-3f51789ca0ad"version = "0.2.1"
[[deps.PyCall]]deps = ["Conda", "Dates", "Libdl", "LinearAlgebra", "MacroTools", "Serialization", "VersionParsing"]git-tree-sha1 = "9816a3826b0ebf49ab4926e2b18842ad8b5c8f04"uuid = "438e738f-606a-5dbb-bf0a-cddfbfd45ab0"version = "1.96.4"[[deps.PyPlot]]deps = ["Colors", "LaTeXStrings", "PyCall", "Sockets", "Test", "VersionParsing"]git-tree-sha1 = "9220a9dae0369f431168d60adab635f88aca7857"uuid = "d330b81b-6aea-500a-939a-2ce795aea3ee"version = "2.11.2"
[[deps.Static]]deps = ["IfElse"]git-tree-sha1 = "f295e0a1da4ca425659c57441bcb59abb035a4bc"uuid = "aedffcd0-7271-4cad-89d0-dc628f76c6d3"version = "0.8.8"[[deps.StaticArrayInterface]]deps = ["ArrayInterface", "Compat", "IfElse", "LinearAlgebra", "PrecompileTools", "Requires", "SparseArrays", "Static", "SuiteSparse"]git-tree-sha1 = "5d66818a39bb04bf328e92bc933ec5b4ee88e436"uuid = "0d7ed370-da01-4f52-bd93-41d350b8b718"version = "1.5.0"weakdeps = ["OffsetArrays", "StaticArrays"][deps.StaticArrayInterface.extensions]StaticArrayInterfaceOffsetArraysExt = "OffsetArrays"StaticArrayInterfaceStaticArraysExt = "StaticArrays"
deps = ["ArgCheck", "BangBang", "ConstructionBase", "InitialValues", "MicroCollections", "Referenceables", "Setfield", "SplittablesBase", "Transducers"]git-tree-sha1 = "34e6bcf36b9ed5d56489600cf9f3c16843fa2aa2"
deps = ["Accessors", "ArgCheck", "BangBang", "ConstructionBase", "InitialValues", "MicroCollections", "Referenceables", "SplittablesBase", "Transducers"]git-tree-sha1 = "70bd8244f4834d46c3d68bd09e7792d8f571ef04"
[[deps.VectorizationBase]]deps = ["ArrayInterface", "CPUSummary", "HostCPUFeatures", "IfElse", "LayoutPointers", "Libdl", "LinearAlgebra", "SIMDTypes", "Static", "StaticArrayInterface"]git-tree-sha1 = "7209df901e6ed7489fe9b7aa3e46fb788e15db85"uuid = "3d5dd08c-fd9d-11e8-17fa-ed2836048c2f"version = "0.21.65"
[[deps.VersionParsing]]git-tree-sha1 = "58d6e80b4ee071f5efd07fda82cb9fbe17200868"uuid = "81def892-9a0e-5fdd-b105-ffc91e053289"version = "1.3.0"